diff --git a/app/agent.py b/app/agent.py
index 04feb23..111388f 100644
--- a/app/agent.py
+++ b/app/agent.py
@@ -46,6 +46,7 @@ def main(prompts: Optional[Tuple[str, ...]] = None) -> None:
 
         user_prompt = prompts[i] if prompts and len(prompts) > i else input(followup_question or default_prompt)
 
+        # Check if done with the chat.
         if user_prompt and user_prompt.lower().strip() in [
             "no",
             "no thanks",