1 files changed, 9 insertions, 13 deletions

diff --git a/src/modules/mining.rs b/src/modules/mining.rs
index 4a1a37f..7a6461a 100644
--- a/src/modules/mining.rs
+++ b/src/modules/mining.rs
@@ -35,30 +35,26 @@ impl Mining {
             }
             _ => (),
         }
-        if self.status == MiningStatus::None {
-            self.start = None;
-        }
     }
 
     pub fn toggle(&mut self) {
-        self.status = match self.status {
-            MiningStatus::None => {
-                self.start = Some(SystemTime::now());
-                MiningStatus::Mining
-            }
-            _ => {
-                self.start = None;
-                MiningStatus::None
-            }
+        match self.status {
+            MiningStatus::None => self.on(),
+            _ => self.off(),
         };
     }
 
+    pub fn on(&mut self) {
+        self.start = Some(SystemTime::now());
+        self.status = MiningStatus::Mining;
+    }
+
     pub fn off(&mut self) {
+        self.start = None;
         self.status = MiningStatus::None;
     }
 
     pub fn take(&mut self) {
         self.status = MiningStatus::Mining;
     }
-
 }