diff --git a/playground/demos/agentic_space_traffic_control/flight_data.py b/playground/demos/agentic_space_traffic_control/flight_data.py
index 3e6a8f19..dd41ff4f 100644
--- a/playground/demos/agentic_space_traffic_control/flight_data.py
+++ b/playground/demos/agentic_space_traffic_control/flight_data.py
@@ -1,6 +1,5 @@
 import requests
 from typing import List, Dict, Any
-from swarms.tools import get_openai_function_schema_from_func
 
 
 def fetch_flights_in_area(
diff --git a/game.py b/playground/demos/agentic_space_traffic_control/game.py
similarity index 94%
rename from game.py
rename to playground/demos/agentic_space_traffic_control/game.py
index 31717a29..aa927f58 100644
--- a/game.py
+++ b/playground/demos/agentic_space_traffic_control/game.py
@@ -69,4 +69,6 @@ flow = AgentRearrange(
     max_loops=3,
 )
 # Run the flow
-flow.run("We're preparing for a launch in Cape canveral, let's begin the launch process, whats the weather like?")
\ No newline at end of file
+flow.run(
+    "We're preparing for a launch in Cape canveral, let's begin the launch process, whats the weather like?"
+)