diff --git a/notebooks/unit2/unit2.ipynb b/notebooks/unit2/unit2.ipynb
index ea78857..a10b711 100644
--- a/notebooks/unit2/unit2.ipynb
+++ b/notebooks/unit2/unit2.ipynb
@@ -771,7 +771,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",
@@ -1748,4 +1748,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}