From 95754dfd8ecf9bbdc021afd7593edf7fbf3e8926 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ond=C5=99ej=20Hru=C5=A1ka?= <ondra@ondrovo.com>
Date: Mon, 12 Feb 2018 09:53:05 +0100
Subject: [PATCH] restart option

---
 pymodoro.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pymodoro.py b/pymodoro.py
index 71e50f6..3e259b4 100644
--- a/pymodoro.py
+++ b/pymodoro.py
@@ -31,6 +31,9 @@ class Pymodoro:
     def on_btn(self, snapshot, timestamp):
         if self.phase == PH_BREAK_OVER:
             self.switch(PH_WORK)
+            
+        elif self.phase == PH_WORK:
+            self.switch(PH_WORK) # restart
 
         elif self.phase == PH_WORK_OVER:
             self.switch(PH_BREAK)