diff --git a/assignment7/utils.py b/assignment7/utils.py
index e9cdc5e0b7b4448f2aa832fa966ef700c1765efa..19b9c4cbb27b994a0f5fa5215eeb5d470a353b82 100644
--- a/assignment7/utils.py
+++ b/assignment7/utils.py
@@ -3,11 +3,11 @@ from typing import List
 
 def format_as_chat(message: str, history: List[List[str]]) -> str:
     """
-    Given a message and a history of previous messages, return a string that formats the conversation as a chat.
+    Given a message and a history of previous messages, returns a string that formats the conversation as a chat.
     Uses the format expected by Meta Llama 3 Instruct.
 
     :param message: A string containing the user's most recent message
-    :param history: A list of lists of strings, where each sublist is a conversation turn:
+    :param history: A list of lists of previous messages, where each sublist is a conversation turn:
         [[user_message1, assistant_reply1], [user_message2, assistant_reply2], ...]
     """
     ...  # TODO