Rebalance

src/main.rs

@@ -144,11 +144,10 @@ impl ScreenState {
 
     fn handle_keyboard(&mut self, message: Event) -> Task<RootMessage> {
         let page = self.stack.last_mut().unwrap();
-        match page {
+        return match page {
             Repetition(page) => page.press(&message),
-            _ => {}
+            _ => Task::none(),
         }
-        Task::none()
     }
 }
 
@@ -202,7 +201,7 @@ trait NavigatedPage<T> {
 }
 
 pub trait KeyPressedPage {
-    fn press(&mut self, message: &keyboard::Event);
+    fn press(&mut self, message: &keyboard::Event) -> Task<RootMessage>;
 }
 
 /*pub fn danger_button(x: &Theme, status: Status) -> Style{