1 files changed, 13 insertions, 6 deletions
diff --git a/2018/src/main.rs b/2018/src/main.rs
index 7e683ca..162a56e 100644
--- a/2018/src/main.rs
+++ b/2018/src/main.rs
@@ -7,10 +7,12 @@ use std::process::exit;
 use argparse::{ArgumentParser, StoreTrue, Store};
 
 mod day1;
+mod day2;
 
 fn day_switch<T: BufRead>(day: i32, reader: T) -> io::Result<()> {
     match day {
         1 => day1::main(reader),
+        2 => day2::main(reader),
         _ => Err(Error::new(ErrorKind::Other, "Invalid day"))
     }
 }
@@ -65,13 +67,18 @@ fn main() -> io::Result<()> {
     }
 
     error_handler(||
-        match day_string.parse::<i32>() {
-            Ok(day) => run_day(day, &options),
-            Err(_) => {
-                for day in 1..1 {
-                    run_day(day, &options)?;
+        if day_string.len() == 0 {
+            for day in 1..3 {
+                run_day(day, &options)?;
+            }
+            Ok(())
+        } else {
+            match day_string.parse::<i32>() {
+                Ok(day) => run_day(day, &options),
+                Err(_) => {
+                    println!("Invalid day argument");
+                    exit(1)
                 }
-                Ok(())
             }
         }
     )