diff --git a/ui/index.html b/ui/index.html
index 9e3517f0..1b697c8e 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -7,7 +7,7 @@
     <link rel="icon" type="image/png" href="/media/images/favicon-32x32.png" sizes="32x32">
     <link rel="stylesheet" href="/media/css/fonts.css?v=1">
     <link rel="stylesheet" href="/media/css/themes.css?v=2">
-    <link rel="stylesheet" href="/media/css/main.css?v=8">
+    <link rel="stylesheet" href="/media/css/main.css?v=9">
     <link rel="stylesheet" href="/media/css/auto-save.css?v=5">
     <link rel="stylesheet" href="/media/css/modifier-thumbnails.css?v=4">
     <link rel="stylesheet" href="/media/css/fontawesome-all.min.css?v=1">
@@ -323,7 +323,7 @@
 <script src="media/js/inpainting-editor.js?v=1"></script>
 <script src="media/js/image-modifiers.js?v=6"></script>
 <script src="media/js/auto-save.js?v=7"></script>
-<script src="media/js/main.js?v=10"></script>
+<script src="media/js/main.js?v=11"></script>
 <script src="media/js/themes.js?v=4"></script>
 <script src="media/js/dnd.js?v=8"></script>
 <script>
diff --git a/ui/media/css/main.css b/ui/media/css/main.css
index 33d18177..d878fa19 100644
--- a/ui/media/css/main.css
+++ b/ui/media/css/main.css
@@ -28,6 +28,7 @@ label {
     font-size: 13px;
     margin-bottom: 6px;
     margin-top: 5px;
+    display: block;
 }
 .image_preview_container {
     margin-top: 10pt;
@@ -656,9 +657,9 @@ input::file-selector-button {
 }
 
 @media (min-width: 700px) {
-    #editor {
+    /* #editor {
         max-width: 480px;
-    }
+    } */
     .float-container {
         padding: 20px;
     }
diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index f6103eba..b368fe27 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -1,7 +1,7 @@
 "use strict" // Opt in to a restricted variant of JavaScript
 const HEALTH_PING_INTERVAL = 5 // seconds
 const MAX_INIT_IMAGE_DIMENSION = 768
-const MIN_GPUS_TO_SHOW_SELECTION = 1
+const MIN_GPUS_TO_SHOW_SELECTION = 2
 
 const IMAGE_REGEX = new RegExp('data:image/[A-Za-z]+;base64')
 
@@ -1460,4 +1460,17 @@ document.querySelectorAll(".tab").forEach(tab => {
     })
 })
 
+window.addEventListener("beforeunload", function(e) {
+    const msg = "Unsaved pictures will be lost!";
+
+    let elementList = document.getElementsByClassName("imageTaskContainer");
+    if (elementList.length != 0) {
+        e.preventDefault();
+        (e || window.event).returnValue = msg;
+        return msg;
+    } else {
+        return true;
+    }
+});
+
 createCollapsibles()