diff --git a/cerebrum/llm/communication.py b/cerebrum/llm/communication.py
index 82fdf09..e111d2d 100644
--- a/cerebrum/llm/communication.py
+++ b/cerebrum/llm/communication.py
@@ -31,13 +31,17 @@ class Response(BaseModel):
     Response class represents the output structure after performing actions.
     
     Attributes:
-        response_message (Optional[str]): The generated response message. Default is None.
-        tool_calls (Optional[List[Dict[str, Any]]]): An optional list of JSON-like objects (dictionaries) 
-            representing the tool calls made during processing. Default is None.
+        response_message (Optional[str]): The generated response message.
+        tool_calls (Optional[List[Dict[str, Any]]]): Tool calls made during processing.
+        finished (bool): Whether the processing is complete.
+        error (Optional[str]): Error message if any.
+        status_code (int): HTTP status code of the response.
     """
-    response_message: Optional[str] = None  # The generated response message, default is None.
-    tool_calls: Optional[List[Dict[str, Any]]] = None  # List of JSON-like objects representing tool calls, default is None.
-    finished: bool
+    response_message: Optional[str] = None
+    tool_calls: Optional[List[Dict[str, Any]]] = None
+    finished: bool = False
+    error: Optional[str] = None
+    status_code: int = 200
 
     class Config:
-        arbitrary_types_allowed = True  # Allows arbitrary types in validation.
\ No newline at end of file
+        arbitrary_types_allowed = True
\ No newline at end of file
diff --git a/llm/communication.py b/llm/communication.py
deleted file mode 100644
index a1afe9a..0000000
--- a/llm/communication.py
+++ /dev/null
@@ -1,12 +0,0 @@
-class Response:
-    def __init__(self, 
-                 response_message=None,
-                 tool_calls=None,
-                 finished=False,
-                 error=None,
-                 status_code=200):
-        self.response_message = response_message
-        self.tool_calls = tool_calls
-        self.finished = finished
-        self.error = error
-        self.status_code = status_code 
\ No newline at end of file