1 files changed, 3 insertions, 1 deletions

diff --git a/src/command.rs b/src/command.rs
index afc00ba..c1a855e 100644
--- a/src/command.rs
+++ b/src/command.rs
@@ -27,7 +27,7 @@ fn call_on_app(s: &mut Cursive, input: &str) {
 
 #[derive(PartialEq)]
 pub enum Command {
-    Add(String, String, Option<u32>), // habit name, habit type, optional goal
+    Add(String, String, Option<u32>, Option<bool>), // habit name, habit type, optional goal, auto tracked
     MonthPrev,
     MonthNext,
     Delete(String),
@@ -50,10 +50,12 @@ impl Command {
                     return Command::Blank;
                 }
                 let goal = args.get(2).map(|g| g.parse::<u32>().ok()).flatten();
+                let auto = args.get(3).map(|g| g.parse::<bool>().ok()).flatten();
                 return Command::Add(
                     args.get_mut(0).unwrap().to_string(),
                     args.get_mut(1).unwrap().to_string(),
                     goal,
+                    auto,
                 );
             }
             "delete" | "d" => {