diff --git a/index.html b/index.html
index a1694df..82ec050 100644
--- a/index.html
+++ b/index.html
@@ -23,7 +23,7 @@
     <select id="modelSelect">
     </select>
     &nbsp;
-    <button id="saveImgBtn">Save Overlay</button>
+    <button id="saveImgBtn">Save Segmentation</button>
     &nbsp;
     <button id="saveSceneBtn">Save Scene</button>
     &nbsp;
diff --git a/main.js b/main.js
index e3eda14..630c51d 100644
--- a/main.js
+++ b/main.js
@@ -377,5 +377,5 @@ async function updateStarCount() {
     console.error("Error fetching star count:", error);
   }
 }
+await main();
 updateStarCount();
-main();