diff --git a/docs/02_notebooks/L5_linear_sweep.ipynb b/docs/02_notebooks/L5_linear_sweep.ipynb
index f60e220..371fe86 100644
--- a/docs/02_notebooks/L5_linear_sweep.ipynb
+++ b/docs/02_notebooks/L5_linear_sweep.ipynb
@@ -109,11 +109,11 @@
     "\n",
     "    plt.xlabel(\"Step\")\n",
     "    plt.ylabel(\"Reward\")\n",
-    "    \n",
+    "\n",
     "    plt.legend([\"Reward\", \"Terminated\"])\n",
-    "    \n",
+    "\n",
     "    if show:\n",
-    "        plt.show()\n"
+    "        plt.show()"
    ]
   },
   {
@@ -163,7 +163,7 @@
    "source": [
     "env_rollout = walk_through_env(env, 100)\n",
     "\n",
-    "plot_rollout_rewards(env_rollout)\n"
+    "plot_rollout_rewards(env_rollout)"
    ]
   },
   {
@@ -209,7 +209,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "projected_env_rollout = walk_through_env(projected_env, 100, render_title=\"Projected labelmap slice\")\n",
+    "projected_env_rollout = walk_through_env(\n",
+    "    projected_env, 100, render_title=\"Projected labelmap slice\",\n",
+    ")\n",
     "plot_rollout_rewards(projected_env_rollout)"
    ]
   },