diff --git a/src/gui.rs b/src/gui.rs
index 95f2c66..725a119 100644
--- a/src/gui.rs
+++ b/src/gui.rs
@@ -48,16 +48,18 @@ impl eframe::App for Omelt {
             }
         });
         egui::CentralPanel::default().show(ctx, |ui| {
-            ui.heading(&self.file_name);
-            if ui.add(egui::Button::new("Save")).clicked() {
-                let _ = self.file.set_len(0);
-                let _ = save_action(&self.text, &mut self.file);
-            }
+            ui.horizontal(|ui| {
+                ui.heading(&self.file_name);
+                if ui.add(egui::Button::new("Save")).clicked() {
+                    let _ = self.file.set_len(0);
+                    let _ = save_action(&self.text, &mut self.file);
+                }
+            });
             egui::ScrollArea::vertical().show(ui, |ui| {
                 //let visible_text = self.text.lines().take(100).collect::<Vec<_>>().join("\n");
                 ui.add_sized(
                     ui.available_size(),
-                    egui::TextEdit::multiline(&mut self.text),
+                    egui::TextEdit::multiline(&mut self.text).lock_focus(true),
                 );
             });
         });
diff --git a/src/main.rs b/src/main.rs
index b395ccc..6ddd9d8 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,7 +1,6 @@
 #![cfg_attr(not(debug_assertions), windows_subsystem = "windows")] // hide console window on Windows in release
 #![allow(rustdoc::missing_crate_level_docs)] // it's an example
 use eframe::egui;
-use eframe::egui::debug_text::print;
 use std::env;
 use std::fs::File;
 use std::fs::OpenOptions;