1 files changed, 8 insertions, 2 deletions

diff --git a/src/command.rs b/src/command.rs
index c4f20fc..f94eab8 100644
--- a/src/command.rs
+++ b/src/command.rs
@@ -17,10 +17,10 @@ fn call_on_app(s: &mut Cursive, input: &str) {
 }
 
 pub enum Command {
-    Add(String, String, Option<u32>),
+    Add(String, String, Option<u32>), // habit name, habit type, optional goal
     MonthPrev,
     MonthNext,
-    Delete,
+    Delete(String),
     Blank,
 }
 
@@ -45,6 +45,12 @@ impl Command {
                     goal,
                 );
             }
+            "delete" | "d" => {
+                if args.len() < 1 {
+                    return Command::Blank;
+                }
+                return Command::Delete(args[0].to_string());
+            }
             "mprev" | "month-prev" => return Command::MonthPrev,
             "mnext" | "month-next" => return Command::MonthNext,
             _ => return Command::Blank,