diff --git a/src/primaite/notebooks/multi-processing.ipynb b/src/primaite/notebooks/multi-processing.ipynb
index fc3ef088..d5b56cbb 100644
--- a/src/primaite/notebooks/multi-processing.ipynb
+++ b/src/primaite/notebooks/multi-processing.ipynb
@@ -17,13 +17,6 @@
     "#!primaite setup"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "Set up training data."
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -64,6 +57,13 @@
     "    cfg = yaml.safe_load(f)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Set up training data."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -78,6 +78,13 @@
     "LEARNING_RATE = 3e-4\n"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Define an environment function."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -108,6 +115,13 @@
     "    return _init\n"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Run experiment."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -135,7 +149,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.12"
+   "version": "3.10.11"
   }
  },
  "nbformat": 4,