diff --git a/docs/tutorials/SKA_forecast.ipynb b/docs/tutorials/SKA_forecast.ipynb
index 95d5cf7..2968e7b 100644
--- a/docs/tutorials/SKA_forecast.ipynb
+++ b/docs/tutorials/SKA_forecast.ipynb
@@ -36,12 +36,11 @@
     "from astropy import units as un\n",
     "from astropy.cosmology import Planck15\n",
     "from matplotlib import colors, rcParams\n",
-    "from scipy.interpolate import RegularGridInterpolator\n",
     "\n",
     "import py21cmsense as p21sense\n",
     "from py21cmsense.observatory import Observatory\n",
     "from py21cmsense.sensitivity import PowerSpectrum\n",
-    "from py21cmsense.theory import EOS2021, TheoryModel\n",
+    "from py21cmsense.theory import EOS2021\n",
     "\n",
     "rcParams.update({\"font.size\": 20})\n",
     "\n",
@@ -436,8 +435,8 @@
     }
    ],
    "source": [
-    "print(\"AA* beam area:%.5f rad2\"% aaast.beam.area.value)  # for 150 MHz (default value)\n",
-    "print(\"AA4 beam area:%.5f rad2\"% aa4.beam.area.value)"
+    "print(\"AA* beam area:%.5f rad2\" % aaast.beam.area.value)  # for 150 MHz (default value)\n",
+    "print(\"AA4 beam area:%.5f rad2\" % aa4.beam.area.value)"
    ]
   },
   {
@@ -1011,14 +1010,31 @@
     }
    ],
    "source": [
-    "plt.plot(ska_aa4_senses2['200.0 MHz']['k_1D_Mpc'], ska_aa4_senses2['200.0 MHz']['sample_and_thermal_1D_mK2'], label='both', color = 'b')\n",
-    "plt.plot(ska_aa4_senses2['200.0 MHz']['k_1D_Mpc'], ska_aa4_senses2['200.0 MHz']['sample_1D_mK2'], label='sample', color = 'b', ls = ':')\n",
-    "plt.plot(ska_aa4_senses2['200.0 MHz']['k_1D_Mpc'], ska_aa4_senses2['200.0 MHz']['thermal_1D_mK2'], label='thermal', color = 'b', ls = '--')\n",
+    "plt.plot(\n",
+    "    ska_aa4_senses2[\"200.0 MHz\"][\"k_1D_Mpc\"],\n",
+    "    ska_aa4_senses2[\"200.0 MHz\"][\"sample_and_thermal_1D_mK2\"],\n",
+    "    label=\"both\",\n",
+    "    color=\"b\",\n",
+    ")\n",
+    "plt.plot(\n",
+    "    ska_aa4_senses2[\"200.0 MHz\"][\"k_1D_Mpc\"],\n",
+    "    ska_aa4_senses2[\"200.0 MHz\"][\"sample_1D_mK2\"],\n",
+    "    label=\"sample\",\n",
+    "    color=\"b\",\n",
+    "    ls=\":\",\n",
+    ")\n",
+    "plt.plot(\n",
+    "    ska_aa4_senses2[\"200.0 MHz\"][\"k_1D_Mpc\"],\n",
+    "    ska_aa4_senses2[\"200.0 MHz\"][\"thermal_1D_mK2\"],\n",
+    "    label=\"thermal\",\n",
+    "    color=\"b\",\n",
+    "    ls=\"--\",\n",
+    ")\n",
     "plt.loglog()\n",
-    "plt.xlabel(r'k [Mpc$^{-1}$]')\n",
-    "plt.ylabel(r'sensitivity [mK$^2$]')\n",
+    "plt.xlabel(r\"k [Mpc$^{-1}$]\")\n",
+    "plt.ylabel(r\"sensitivity [mK$^2$]\")\n",
     "plt.legend(frameon=False)\n",
-    "plt.text(0.85, 0.1, 'AA4', transform=plt.gca().transAxes, fontsize=20, verticalalignment='bottom')\n",
+    "plt.text(0.85, 0.1, \"AA4\", transform=plt.gca().transAxes, fontsize=20, verticalalignment=\"bottom\")\n",
     "plt.show()"
    ]
   },