diff --git a/main.py b/main.py
index 4f6147b..65dbd07 100644
--- a/main.py
+++ b/main.py
@@ -175,22 +175,7 @@ def run_demo() -> None:
     print("Code under review:")
     print(initial_state["code"])
 
-    final_state = initial_state.copy()
-    for chunk in graph.stream(initial_state, stream_mode="updates"):
-        for node_name, update in chunk.items():
-            final_state.update(update)
-            print()
-            if node_name == "draft_review":
-                print("--- Draft Review ---")
-                print(final_state["draft_review"])
-            elif node_name == "reflect":
-                print("--- Critic Scores ---")
-                print(final_state["criteria_scores"])
-                print(f"Weakest criterion: {final_state['weakest_criterion']}")
-                print(f"Verdict: {final_state['verdict']}")
-            elif node_name == "rewrite":
-                print(f"--- Rewritten Review After Round {final_state['round']} ---")
-                print(final_state["draft_review"])
+    final_state = graph.run(initial_state)
 
     print()
     print("=== Final Result ===")