diff --git a/notebooks/Building a simple agent.ipynb b/notebooks/Building a simple agent.ipynb
index 2e4308b7..66798e8a 100644
--- a/notebooks/Building a simple agent.ipynb	
+++ b/notebooks/Building a simple agent.ipynb	
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/MarcCote/TextWorld/blob/msr_summit_2021/notebooks/Building%20a%20simple%20agent.ipynb)"
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/microsoft/TextWorld/blob/main/notebooks/Building%20a%20simple%20agent.ipynb)"
    ]
   },
   {
diff --git a/notebooks/Handcrafting a game.ipynb b/notebooks/Handcrafting a game.ipynb
index 058da0ef..5485dbaa 100644
--- a/notebooks/Handcrafting a game.ipynb	
+++ b/notebooks/Handcrafting a game.ipynb	
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/MarcCote/TextWorld/blob/msr_summit_2021/notebooks/Handcrafting%20a%20game.ipynb)"
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/microsoft/TextWorld/blob/main/notebooks/Handcrafting%20a%20game.ipynb)"
    ]
   },
   {
@@ -31,6 +31,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# For Google Colab, first install\n",
+    "#!apt install chromium-chromedriver\n",
+    "#!pip install selenium==3.12.0\n",
+    "# then\n",
     "!pip install textworld[vis]"
    ]
   },
diff --git a/notebooks/Playing TextWorld generated games with OpenAI Gym.ipynb b/notebooks/Playing TextWorld generated games with OpenAI Gym.ipynb
index 0741097d..b7e5b057 100644
--- a/notebooks/Playing TextWorld generated games with OpenAI Gym.ipynb	
+++ b/notebooks/Playing TextWorld generated games with OpenAI Gym.ipynb	
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/MarcCote/TextWorld/blob/msr_summit_2021/notebooks/Playing%20TextWorld%20generated%20games%20with%20OpenAI%20Gym.ipynb)"
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/microsoft/TextWorld/blob/main/notebooks/Playing%20TextWorld%20generated%20games%20with%20OpenAI%20Gym.ipynb)"
    ]
   },
   {
@@ -2350,7 +2350,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3.8.10 64-bit",
    "language": "python",
    "name": "python3"
   },
@@ -2364,7 +2364,12 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.3"
+   "version": "3.8.10"
+  },
+  "vscode": {
+   "interpreter": {
+    "hash": "31f2aee4e71d21fbe5cf8b01ff0e069b9275f58929596ceb00d14d90e3e16cd6"
+   }
   }
  },
  "nbformat": 4,
diff --git a/notebooks/Playing text-based games with TextWorld.ipynb b/notebooks/Playing text-based games with TextWorld.ipynb
index 6fd4a0a3..f7586791 100644
--- a/notebooks/Playing text-based games with TextWorld.ipynb	
+++ b/notebooks/Playing text-based games with TextWorld.ipynb	
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/MarcCote/TextWorld/blob/msr_summit_2021/notebooks/Playing%20text-based%20games%20with%20TextWorld.ipynb)"
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/microsoft/TextWorld/blob/main/notebooks/Playing%20text-based%20games%20with%20TextWorld.ipynb)"
    ]
   },
   {
@@ -63,6 +63,23 @@
     "!echo \"Done.\""
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "**[Optional]** On Google Colab, you need to download additional data beforehand."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!wget https://aka.ms/textworld/notebooks/data.zip\n",
+    "!unzip -nq data.zip && rm -f data.zip"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -659,7 +676,7 @@
     }
    ],
    "source": [
-    "textworld.play(\"./games/rewardsSparse_goalNone.z8\")"
+    "textworld.play(\"./games/tw-rewardsSparse_goalNone.z8\")"
    ]
   },
   {