diff --git a/notebooks/unit2/requirements-unit2.txt b/notebooks/unit2/requirements-unit2.txt
index 733afc8..995b24b 100644
--- a/notebooks/unit2/requirements-unit2.txt
+++ b/notebooks/unit2/requirements-unit2.txt
@@ -8,3 +8,4 @@ pyyaml==6.0
 imageio
 imageio_ffmpeg
 pyglet==1.5.1
+tqdm
diff --git a/notebooks/unit2/unit2.ipynb b/notebooks/unit2/unit2.ipynb
index 1a075ae..5555554 100644
--- a/notebooks/unit2/unit2.ipynb
+++ b/notebooks/unit2/unit2.ipynb
@@ -280,6 +280,7 @@
         "import random\n",
         "import imageio\n",
         "import os\n",
+        "import tqdm\n",
         "\n",
         "import pickle5 as pickle\n",
         "from tqdm.notebook import tqdm"
@@ -771,7 +772,7 @@
       "outputs": [],
       "source": [
         "def train(n_training_episodes, min_epsilon, max_epsilon, decay_rate, env, max_steps, Qtable):\n",
-        "  for episode in range(n_training_episodes):\n",
+        "  for episode in tqdm(range(n_training_episodes)):\n",
         "    # Reduce epsilon (because we need less and less exploration)\n",
         "    epsilon = min_epsilon + (max_epsilon - min_epsilon)*np.exp(-decay_rate*episode)\n",
         "    # Reset the environment\n",
@@ -1747,4 +1748,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}