diff --git a/reinforcement_learning.py b/reinforcement_learning.py
index 8549a051c7e87aa185df59e14aae806d26b2a715..b2ff13390f24d3a1a3a6abb2155cbf37a9ec6747 100644
--- a/reinforcement_learning.py
+++ b/reinforcement_learning.py
@@ -1,3 +1,4 @@
+from IPython.display import clear_output
 import time
 
 def animate(frames, delay_per_frame):