1 files changed, 16 insertions, 5 deletions
diff --git a/src/main.rs b/src/main.rs
index 9bf6c51..2872fcf 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,13 +3,14 @@ use iced::{
     Element, Sandbox, Settings,
 };
 
+use volute::volume::{CubicMetre, Volume};
+
 pub fn main() -> iced::Result {
     App::run(Settings::default())
 }
 
-#[derive(Default)]
 struct App {
-    displacement: f64,
+    displacement: Box<dyn Volume>,
     ui: UI,
 }
 
@@ -21,11 +22,21 @@ struct UI {
 impl App {
     fn set_displacement(&mut self, displacement: &str) {
         if displacement.len() == 0 {
-            self.displacement = 0.;
+            self.displacement.set(0.0);
+            self.ui.displacement = "".to_string();
         } else if let Ok(d) = displacement.parse::<f64>() {
-            self.displacement = d;
+            self.displacement.set(d);
+            self.ui.displacement = String::from(displacement);
+        }
+    }
+}
+
+impl Default for App {
+    fn default() -> Self {
+        Self {
+            displacement: Box::new(CubicMetre::default()),
+            ui: UI::default(),
         }
-        self.ui.displacement = String::from(displacement);
     }
 }