diff --git a/src/main.rs b/src/main.rs
index c883fbd..26e1da6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,10 +1,10 @@
 extern crate cat2text;
-use cat2text::base4;
-use std::io::{self, stdout, Write};
+use cat2text::{base4, core};
+use std::{io, process::exit};
 
 fn main() {
     let stdin = io::stdin();
-    let mut input;
+    let mut input = String::new();
 
     loop {
         println!("Pick your translation:");
@@ -12,8 +12,6 @@ fn main() {
         println!("2) text to cat");
         input = "".to_string();
         stdin.read_line(&mut input).unwrap();
-        print!("~> ");
-        stdout().flush().unwrap();
         let trimmed = input.trim();
         if trimmed == "1".to_string() {
             input = "".to_string();