fix bug with empty countview entries

1 files changed, 4 insertions, 2 deletions

diff --git a/src/habit.rs b/src/habit.rs
index 88e1cb6..967e425 100644
--- a/src/habit.rs
+++ b/src/habit.rs
@@ -50,12 +50,14 @@ impl Habit<bool> {
 impl Habit<u32> {
     pub fn increment(&mut self, date: NaiveDate) {
         if let Some(v) = self.stats.get_mut(&date) {
-            *v += 1
+            *v += 1;
+        } else {
+            self.insert_entry(date, 1);
         }
     }
     pub fn decrement(&mut self, date: NaiveDate) {
         if let Some(v) = self.stats.get_mut(&date) {
-            if *v >= 1 {
+            if *v > 0 {
                 *v -= 1;
             } else {
                 *v = 0;