diff --git a/UI_V2/pages/home/home.py b/UI_V2/pages/home/home.py
index ef9c694..abc3b69 100644
--- a/UI_V2/pages/home/home.py
+++ b/UI_V2/pages/home/home.py
@@ -199,8 +199,7 @@ class Home:
                     self.chat_header,
                     ft.Divider(),
                     ft.Container(
-                        height=280, #if self.page.width > 500 else None,  # chat messages area height
-                        #expand = True if self.page.width < 500 else None,
+                        height=280 if self.page.width > 500 else 200,  
                         content=self.messages_list,
                     ),
                     ft.Divider(),
@@ -424,7 +423,7 @@ class Home:
 
         self.chat_input.value = ""
         # Keep the focus in the chat input so the user can continue typing
-        #self.chat_input.focus()
+        self.chat_input.focus()
         self.chat_input.update()
         self.page.update()