diff --git a/CS3.1.py b/CS3.1.py
index 66d69bb..0899e19 100644
--- a/CS3.1.py
+++ b/CS3.1.py
@@ -83,14 +83,14 @@ def handle_socket_message(data, addr, conn):
         action = data.get('type')
         if action == 'register':
             result = register_user(data.get('username'), data.get('password'))
-            return {"status": "success" if result['success'] else "error", "message": result['message']}
+            return {"type": "register","status": "success" if result['success'] else "error", "message": result['message']}
         elif action == 'login':
             if isuserxist(data['username']):
                 if ispsswdright(data['username'], data['password']):
                     active_users[addr[0]] = data['username']
                     chat_connections.append(conn)
-                    return {"status": "success", "message": "Login successful"}
-            return {"status": "error", "message": "Invalid credentials"}
+                    return {"type": "login", "status": "success", "message": "Login successful"}
+            return {"type": "login", "status": "error", "message": "Invalid credentials"}
         elif action == 'chat':
             if addr[0] in active_users:
                 message = {
@@ -99,8 +99,8 @@ def handle_socket_message(data, addr, conn):
                     "message": data['message']
                 }
                 broadcast_message(message)
-                return {"status": "success"}
-            return {"status": "error", "message": "Not logged in"}
+                return {"type": "chat", "status": "success", }
+            return {"type": "chat", "status": "error", "message": "Not logged in"}
     except Exception as e:
         return {"status": "error", "message": str(e)}