diff --git a/app/components/Header.tsx b/app/components/Header.tsx
index 0445db4..8a930e3 100644
--- a/app/components/Header.tsx
+++ b/app/components/Header.tsx
@@ -43,7 +43,7 @@ const Header: React.FC<HeaderProps> = ({ onViewChange, showDivs, toggleDivs, sho
         <div className="header-button header-logo">
           {/* AI logo or text */}
         </div>
-        <div className="login-button">
+        <div className="login-button-container">
           <Login /> 
         </div>
       </header>
diff --git a/app/components/InputFrontend.tsx b/app/components/InputFrontend.tsx
index 74539e6..f02442b 100644
--- a/app/components/InputFrontend.tsx
+++ b/app/components/InputFrontend.tsx
@@ -40,7 +40,7 @@ const InputFrontend = React.forwardRef<HTMLDivElement, InputProps>(
         <button type="button" onClick={() => onSendClick(inputValue, false)} disabled={inputDisabled ? true : false}>
           <img src="/img/send.svg" alt="send" />
         </button>
-        <button className={`microphone-button ${isRecording ? "red":"green"}`} type="button" onClick={onMicClick}>
+        <button className={`microphone-button ${isRecording ? "red": "var(--input-button-color)"}`} type="button" onClick={onMicClick}>
           <img src="/img/microphone.svg" alt="microphone" />
         </button>
       </div>
diff --git a/app/components/Models.tsx b/app/components/Models.tsx
index 17f7f46..894623a 100644
--- a/app/components/Models.tsx
+++ b/app/components/Models.tsx
@@ -1,3 +1,4 @@
+"use client";
 import React, { useState, useEffect } from 'react';
 
 // Define the available model options
@@ -18,9 +19,10 @@ const modelDropdown = {
 const Models: React.FC = () => {
   // Initialize state with value from localStorage or default to ''
   const [selectedModel, setSelectedModel] = useState('');
-  const radioSelection = localStorage.getItem('radioSelection')
-
+  const [radioSelection, setRadioSelection] = useState<string | null>("")
+  
   useEffect(() => {
+    setRadioSelection(localStorage.getItem('radioSelection'))
     const handleStorageChange = () => {
       setSelectedModel(localStorage.getItem('selectedModel') || '');
     };
diff --git a/app/styles/header.css b/app/styles/header.css
index 5e84cc5..99343f2 100644
--- a/app/styles/header.css
+++ b/app/styles/header.css
@@ -40,16 +40,23 @@ header{
 .nav-links{
     position: absolute;
     display: flex;
-    top: 2vh;
     gap: 1em;
-    width: 25em;
-    height: 2em;
+    width: 35vw;
+    height: 100%;
+    align-items: center;
 }
 
 .nav-btn{
-    background-color: transparent;
+    background-color: var(--input-button-color);
     border: none;
     font-size: 0.9em;
+    height: 50%;
+    border-radius: 5px;
+    padding: 1px 15px;
+}
+
+.nav-btn:hover{
+    background-color: var(--input-button-hover-color);
 }
 
 .header-logo{
@@ -64,7 +71,13 @@ header{
     border: none;
 }
 
-.login-button .header-login-button{
+.login-button-container{
+    height: 100%;
+    display: flex;
+    align-items: center;
+}
+
+.header-login-button{
     font-size: 2vh;
     position: absolute;
     top: 1.5vh;
@@ -78,6 +91,6 @@ header{
     transition: background-color 0.3s;
 }
 
-.login-button button:hover {
+.header-login-button:hover {
     background-color: var(--input-button-hover-color);
 }
\ No newline at end of file