diff --git a/scripts/ai-heatmap.ipynb b/scripts/ai-heatmap.ipynb
index 3a4a0a2..f2abada 100644
--- a/scripts/ai-heatmap.ipynb
+++ b/scripts/ai-heatmap.ipynb
@@ -15,10 +15,16 @@
     "\n",
     "\n",
     "2. **Compare your ChatGPT usage with Claude usage.**\n",
-    "It's interesting to see when you start switching from one to another (thanks Chris Alexiuk!)\n",
-    "\n",
-    "![ChatGPT vs. Claude usage](chatgpt-claude.png)\n",
+    "It's interesting to see when you start switching from one to another (this visualization is from Chris Alexiuk!)\n",
     "\n",
+    "![ChatGPT vs. Claude usage](chatgpt-claude.png)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "6f66535c",
+   "metadata": {},
+   "source": [
     "## 1. Visualize ChatGPT usage\n",
     "1. Export your ChatGPT conversations. From your ChatGPT account, go to Settings -> Data controls -> Export\n",
     "2. Unzip the data export\n",