diff --git a/Section #4 - Capstone/simpsons.ipynb b/Section #4 - Capstone/simpsons.ipynb
index 3d8e127..02d9e71 100644
--- a/Section #4 - Capstone/simpsons.ipynb	
+++ b/Section #4 - Capstone/simpsons.ipynb	
@@ -273,6 +273,7 @@
       "source": [
         "# Training the model\n",
         "\n",
+        "import scipy\n",
         "from tensorflow.keras.callbacks import LearningRateScheduler\n",
         "callbacks_list = [LearningRateScheduler(canaro.lr_schedule)]\n",
         "training = model.fit(train_gen,\n",
@@ -368,4 +369,4 @@
       }
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/Section #4 - Capstone/simpsons.py b/Section #4 - Capstone/simpsons.py
index d96fce4..34f4259 100644
--- a/Section #4 - Capstone/simpsons.py	
+++ b/Section #4 - Capstone/simpsons.py	
@@ -13,6 +13,7 @@
 import matplotlib.pyplot as plt
 from tensorflow.keras.utils import to_categorical
 from tensorflow.keras.callbacks import LearningRateScheduler
+import scipy
 
 IMG_SIZE = (80,80)
 channels = 1