diff --git a/client/index.html b/client/index.html
index 22365a1..64f2165 100644
--- a/client/index.html
+++ b/client/index.html
@@ -2,9 +2,9 @@
 <html lang="en">
   <head>
     <meta charset="UTF-8" />
-    <link rel="icon" type="image/svg+xml" href="/vite.svg" />
+    <link rel="icon" type="image/svg+xml" href="/icon.ico" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Vite + React</title>
+    <title>Code Editor</title>
   </head>
   <body>
     <div id="root"></div>
diff --git a/client/public/icon.ico b/client/public/icon.ico
new file mode 100644
index 0000000..24c55b6
Binary files /dev/null and b/client/public/icon.ico differ
diff --git a/client/src/pages/ViewCode.tsx b/client/src/pages/ViewCode.tsx
index 95be77b..ef836a4 100644
--- a/client/src/pages/ViewCode.tsx
+++ b/client/src/pages/ViewCode.tsx
@@ -25,7 +25,9 @@ export default function ViewCode() {
     );
   };
 
-  Prism.highlightAll();
+  useEffect(() => {
+    Prism.highlightAll();
+  }, [code]);
 
   return (
     <div className="editor">
