diff --git a/src/entrypoints/background/index.ts b/src/entrypoints/background/index.ts
index 84bd53bc..8ea2e809 100644
--- a/src/entrypoints/background/index.ts
+++ b/src/entrypoints/background/index.ts
@@ -8,18 +8,19 @@ import { captureScreenListener } from './sidebar/captureScreenListener'
 import { sendSidebarShortcut } from './sidebar/sendSidebarShortcut'
 import { sidebarToggleListeners } from './sidebar/sidebarToggleListeners'
 
-backgroundLog()
+export default defineBackground({
+  main() {
+    // Initialize logging
+    backgroundLog()
 
-// =========================== //
-// Sidebar Scripts
-// =========================== //
-sidebarToggleListeners()
-sendSidebarShortcut()
-captureScreenListener()
+    // Sidebar functionality
+    sidebarToggleListeners()
+    sendSidebarShortcut()
+    captureScreenListener()
 
-// =========================== //
-// Quick menu Scripts
-// =========================== //
-createContextMenu()
-forwardContextMenuClicks()
-createContextMenuOnStorageChange()
+    // Quick menu functionality
+    createContextMenu()
+    forwardContextMenuClicks()
+    createContextMenuOnStorageChange()
+  },
+})
diff --git a/images/icon-128.png b/src/public/icon-128.png
similarity index 100%
rename from images/icon-128.png
rename to src/public/icon-128.png
diff --git a/images/icon-16.png b/src/public/icon-16.png
similarity index 100%
rename from images/icon-16.png
rename to src/public/icon-16.png
diff --git a/images/icon-32.png b/src/public/icon-32.png
similarity index 100%
rename from images/icon-32.png
rename to src/public/icon-32.png
diff --git a/images/icon-48.png b/src/public/icon-48.png
similarity index 100%
rename from images/icon-48.png
rename to src/public/icon-48.png
diff --git a/images/robot.png b/src/public/robot.png
similarity index 100%
rename from images/robot.png
rename to src/public/robot.png
diff --git a/vite.config.ts b/vite.config.ts
index 7b798060..643b3db2 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -9,13 +9,5 @@ export default defineConfig({
         port: 5173,
       },
     },
-    build: {
-      rollupOptions: {
-        input: {
-          sidebar: 'src/pages/sidebar/index.html',
-          settings: 'src/pages/settings/index.html',
-        },
-      },
-    },
   }),
 })