diff --git a/demo/ball_bounces.py b/demo/ball_bounces.py
index 2d8df59cb77a47f33d53a8c4983fd87bc9640e07..0dcd8fa0542d9180f00c9590572431b516ddc0b8 100644
--- a/demo/ball_bounces.py
+++ b/demo/ball_bounces.py
@@ -85,6 +85,7 @@ if __name__ == "__main__":
     # Longer term: We should initialize by writing expressions like "-2 \leq myball1.x \leq 5"
     # "-2 \leq myball1.x + myball2.x \leq 5"
     traces = bouncingBall.simulate(10)
+    # There should be a print({traces}) function
     fig = go.Figure()
     fig = plotly_simulation_anime(traces, tmp_map, fig)
     fig.show()
diff --git a/dryvr_plus_plus/example/example_agent/ball_agent.py b/dryvr_plus_plus/example/example_agent/ball_agent.py
index 31664199e62872a5605fcce0a6aab872e06407bb..9fd550e84538fa409b3cbf8220ce194d9819a57e 100644
--- a/dryvr_plus_plus/example/example_agent/ball_agent.py
+++ b/dryvr_plus_plus/example/example_agent/ball_agent.py
@@ -51,4 +51,6 @@ class BallAgent(BaseAgent):
         return np.array(trace)
 
 if __name__ == '__main__':
-    aball = BallAgent('red_ball',file_name="/Users/mitras/Dpp/GraphGeneration/demo/ball_bounces.py")
\ No newline at end of file
+    aball = BallAgent('red_ball',file_name="/Users/mitras/Dpp/GraphGeneration/demo/ball_bounces.py")
+    trace = aball.TC_simulate({'none'},[5, 10, 2, 2],10)
+    print(trace)
\ No newline at end of file