diff --git a/examples/storm/storm.ipynb b/examples/storm/storm.ipynb
index 8b393c63c..d56ca5f98 100644
--- a/examples/storm/storm.ipynb
+++ b/examples/storm/storm.ipynb
@@ -679,16 +679,6 @@
     ").with_config(run_name=\"GenerateAnswer\")"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "#### Reference Store\n",
-    "\n",
-    "The research process uncovers a large number of reference documents that we may want to query during the final article-writing process.\n",
-    "Here, we will createa multi-vector retriever and store all the searched documents inline."
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 19,
@@ -1028,7 +1018,7 @@
     "\n",
     "#### Create Retriever\n",
     "\n",
-    "When completing a given section of the outline, the LLM can retrieve reference documents unearthed during the conversations above.\n",
+    "The research process uncovers a large number of reference documents that we may want to query during the final article-writing process.\n",
     "\n",
     "First, create the retriever:"
    ]
@@ -1336,7 +1326,9 @@
     "3. Refine the outline based on the conversations\n",
     "4. Index the reference docs from the conversations\n",
     "5. Write the individual sections of the article\n",
-    "6. Write the final wiki"
+    "6. Write the final wiki\n",
+    "\n",
+    "The state tracks the outputs of each stage."
    ]
   },
   {
@@ -1466,6 +1458,13 @@
     "    }"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "#### Create the graph"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 73,