More work on partition stage.

src/main.rs

@@ -34,6 +34,7 @@ use crate::stages::welcome::WelcomeStage;
 rust_i18n::i18n!("src/locales", fallback = "en");
 mod kira_theming;
 mod kira_scroll_list;
+//mod kira_disk_layout;
 mod stage;
 mod stages;
 
@@ -230,8 +231,16 @@ fn update(k_state: &mut KiraState, message: Message) -> Task<Message> {
             if let Views::Keyboard(keyboard_view) = &mut k_state.current_view {
                 match keyboard_view.update(keyboard_msg) {
                     StageAction::Next(keyboard_res) => {
-                        k_state.config.config_trail.push(keyboard_res);
-                        k_state.current_view = Views::Partition(stages::partition::PartitionStage::new());
+                        match stages::partition::PartitionStage::new(&k_state.toml_config) {
+                            Ok(part_stage) => {
+                                k_state.config.config_trail.push(keyboard_res);
+                                k_state.current_view = Views::Partition(part_stage);
+                            },
+                            Err(ex) => {
+                                println!("Error, cannot init partition stage: {}", ex);
+                            }
+                            
+                        }
                         Task::none()
                     }
                     StageAction::Back => {