diff --git a/README.md b/README.md
index cdffeda1bcf601c81c8732e5c7ce3f9f820e2aeb..3413b6dda9ebcb2b5a4aa6b6bde56c5427a43639 100644
--- a/README.md
+++ b/README.md
@@ -104,4 +104,4 @@ In this case, no decision could be made as we had no matches — so our decision
 
 |     |     |
 | --- | --- |
-| 🏃 Walkthrough | Quickstart Python notebook |
\ No newline at end of file
+| 🏃 [Walkthrough](https://colab.research.google.com/github/aurelio-labs/semantic-router/blob/main/walkthrough.ipynb) | Quickstart Python notebook |
\ No newline at end of file
diff --git a/walkthrough.ipynb b/walkthrough.ipynb
index 1b100e867e3e58d9f8a5c6df51f044463769e34e..81bb3ec270a090b699c633d272a40d20bf25c19e 100644
--- a/walkthrough.ipynb
+++ b/walkthrough.ipynb
@@ -107,11 +107,11 @@
    "source": [
     "from semantic_router.encoders import CohereEncoder\n",
     "from getpass import getpass\n",
-    "\n",
     "import os\n",
     "\n",
-    "# os.environ[\"COHERE_API_KEY\"] = getpass(\"Enter Cohere API Key: \")\n",
-    "os.environ[\"COHERE_API_KEY\"]\n",
+    "os.environ[\"COHERE_API_KEY\"] = os.environ[\"COHERE_API_KEY\"] or \\\n",
+    "    getpass(\"Enter Cohere API Key: \")\n",
+    "\n",
     "encoder = CohereEncoder()"
    ]
   },