diff --git a/.DS_Store b/.DS_Store
new file mode 100644
index 0000000..adac3bf
Binary files /dev/null and b/.DS_Store differ
diff --git a/scripts/ai-heatmap.ipynb b/scripts/ai-heatmap.ipynb
index c006cbf..3a4a0a2 100644
--- a/scripts/ai-heatmap.ipynb
+++ b/scripts/ai-heatmap.ipynb
@@ -5,7 +5,21 @@
    "id": "2a305a33",
    "metadata": {},
    "source": [
-    "## How to use this script\n",
+    "Visualize your ChatGPT and Claude's usage.\n",
+    "\n",
+    "This has two scripts:\n",
+    "\n",
+    "1. **Visualize only ChatGPT usage.**\n",
+    "\n",
+    "![ChatGPT usage](chatgpt.png)\n",
+    "\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",
+    "\n",
+    "## 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",
     "3. Point `convo_folder` to your exported data folder\n",
@@ -178,7 +192,7 @@
    "id": "1c5de1c0",
    "metadata": {},
    "source": [
-    "### Adding Claude \n",
+    "## 2. Visualize ChatGPT vs. Claude usage\n",
     "\n",
     "You can optionally add your Claude data as well, through the following step:\n",
     "\n",
@@ -398,7 +412,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": ".venv",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -412,7 +426,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.11"
+   "version": "3.11.5"
   }
  },
  "nbformat": 4,
diff --git a/scripts/chatgpt-claude.png b/scripts/chatgpt-claude.png
new file mode 100644
index 0000000..767c91d
Binary files /dev/null and b/scripts/chatgpt-claude.png differ
diff --git a/scripts/chatgpt.png b/scripts/chatgpt.png
new file mode 100644
index 0000000..59ce27f
Binary files /dev/null and b/scripts/chatgpt.png differ