1 files changed, 9 insertions, 9 deletions

diff --git a/src/main.rs b/src/main.rs
index f3f9d2d..0cd192d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,20 +11,12 @@ fn main() {
     loop {
         let mut input = String::new();
         stdin().read_line(&mut input).unwrap();
-
         let input = input.trim();
         let input = input.replace(" ", "");
-        let input = autobalance_parens(&input[..]).unwrap();
-
         if input == "exit" {
             return
         }
-
-        let lexed = lexer(&input[..]);
-        let postfixed = to_postfix(lexed.unwrap());
-        let evaled = eval_postfix(postfixed.unwrap());
-        println!("ans: {}", evaled.unwrap());
-        println!();
+        println!("ans: {}\n", eval_math_expression(&input[..]).unwrap());
     }
 }
 
@@ -50,3 +42,11 @@ fn autobalance_parens(input: &str) -> Result<String, String> {
         Ok(balanced)
     }
 }
+
+fn eval_math_expression(input: &str) -> Result<f64, String> {
+    let input     = autobalance_parens(&input[..])?;
+    let lexed     = lexer(&input[..])?;
+    let postfixed = to_postfix(lexed)?;
+    let evaled    = eval_postfix(postfixed)?;
+    Ok(evaled)
+}