Added a close function #18

src/extensions/input_handling.rs

@@ -15,6 +15,10 @@ impl Extension for InputExtension {
             }
         });
     }
+
+    fn on_close(&mut self, _screen: std::sync::Arc<crate::Screen>) {
+        crossterm::terminal::disable_raw_mode().unwrap();
+    }
 }
 
 impl crate::extensions::Event for crossterm::event::Event {

src/extensions/mod.rs

@@ -22,6 +22,7 @@ use crate::{
 
 pub trait Extension {
     fn init(&mut self, _screen: Arc<Screen>) {}
+    fn on_close(&mut self, _screen: Arc<Screen>) {}
     fn render_widget(&mut self, _scope: &mut RenderScope, _widget: &Arc<Widget>) {}
 }