diff --git a/ahfe_tutorial/python_tutorial.ipynb b/ahfe_tutorial/python_tutorial.ipynb
index c7105c7..3e1d239 100644
--- a/ahfe_tutorial/python_tutorial.ipynb
+++ b/ahfe_tutorial/python_tutorial.ipynb
@@ -624,7 +624,15 @@
    "id": "7f1dfcff-bb06-4508-9f16-4e3de18a5f93",
    "metadata": {},
    "source": [
-    "## 4. Running the AHFE simulation using the CLI command `openfe quickrun`"
+    "## 4. Running the AHFE simulation"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "fe618563-c445-4fb3-bc0c-6c08148ddede",
+   "metadata": {},
+   "source": [
+    "### (a) Using the CLI"
    ]
   },
   {
@@ -689,10 +697,17 @@
   },
   {
    "cell_type": "markdown",
-   "id": "318ff872",
+   "id": "bc919dc3-1281-49d2-ad92-4a3355d8bbe5",
+   "metadata": {},
+   "source": [
+    "### (b) Using the Python API"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "21b64394-fdc7-47e9-aadd-488a55c1d0bb",
    "metadata": {},
    "source": [
-    "## 5. Running the AHFE simulation using the Python API\n",
     "**Creating the `ProtocolDAG`**\n",
     "\n",
     "Once we have the two `ChemicalSystem`s, and the `Protocol`, we can create the `ProtocolDAG`.\n",