diff --git a/docs/02_notebooks/01_working_with_penai.ipynb b/docs/02_notebooks/01_working_with_penai.ipynb
index ad80034..7cf5afb 100644
--- a/docs/02_notebooks/01_working_with_penai.ipynb
+++ b/docs/02_notebooks/01_working_with_penai.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -12,7 +12,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -20,7 +20,6 @@
     "from IPython.display import SVG, display\n",
     "\n",
     "from penai.hierarchy_generation.utils import InteractivePenpotDesignVisualizer\n",
-    "from penai.models import PenpotProject\n",
     "from penai.registries.projects import SavedPenpotProject\n",
     "from penai.render import WebDriverSVGRenderer\n",
     "from penai.utils.vis import (\n",
@@ -73,11 +72,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Load a project from a project folder\n",
-    "project = PenpotProject.from_directory(\"../../data/raw/designs/Material Design 3\")\n",
+    "# Load a project from the example project registry\n",
+    "project = SavedPenpotProject.MATERIAL_DESIGN_3.load(pull=True)\n",
     "\n",
-    "# OR load a project from the example project registry\n",
-    "project = SavedPenpotProject.MATERIAL_DESIGN_3.load(pull=True)"
+    "# OR from a project folder\n",
+    "# project = PenpotProject.from_directory(\"../../data/raw/designs/Material Design 3\")"
    ]
   },
   {