diff --git a/week-3/color-picker.html b/week-3/color-picker.html
index f2d558d..c1659f2 100644
--- a/week-3/color-picker.html
+++ b/week-3/color-picker.html
@@ -9,31 +9,28 @@
 <body>
 
     <form >
-
         <input type="color" name="bg" id="bg">
     </form>
     
 
     <script>
-    let picker = document.getElementById('bg');
+    let picker = document.querySelector('#bg');
     let body = document.querySelector('body');
-    let keyName = "bgColor";
-    let bgColor = window.localStorage.getItem(keyName) || '#ffffff';
 
+  
     picker.addEventListener('change', function(event) {
         let hex = this.value;
-        console.log(hex);
-        setBg(hex);
-        window.localStorage.setItem(keyName, hex);
 
-    })
+        body.style.backgroundColor = hex;
+        localStorage.setItem("myBgColor", hex);
+
+    });
 
-    function setBg(color){
-        body.style.backgroundColor = color;
-        picker.value = color;
-    }
 
-    setBg(bgColor);
+    // Get the current color
+    let bgColor = localStorage.getItem("myBgColor") || '#cccccc';
+    body.style.backgroundColor = bgColor;
+    picker.value = bgColor;
 
 
     </script>
