diff --git a/main.py b/main.py
index 5a9bd73..6007edf 100644
--- a/main.py
+++ b/main.py
@@ -30,15 +30,18 @@ def main():
         with_code=True,
         commit_message="Add more functionality to signature description parsing",
     )
-    """
-    agent = AutoProgram.from_precompiled("farouk1/claude-code", signature=ClaudeCodeSignature, working_directory=".", permission_mode="acceptEdits", allowed_tools=["Read", "Write", "Bash"])
+    # agent = AutoProgram.from_precompiled("farouk1/claude-code", signature=ClaudeCodeSignature, working_directory=".", permission_mode="acceptEdits", allowed_tools=["Read", "Write", "Bash"])
     
     # Test the agent
-    result = agent(message="create a python program that prints 'Hello, World!' and save it to a file in this directory")
+    result = cc(message="create a python program that prints 'Hello, World!' and save it to a file in this directory")
     print(result.output.files)
     print(result.output)
     print(result.usage)
-    """
+
+    print()
+    print("Trace:")
+    for i, item in enumerate(result.trace):
+        print(f"  {i}: {item}")
 
 
 if __name__ == "__main__":