diff --git a/notebooks/unit5/unit5.ipynb b/notebooks/unit5/unit5.ipynb
index 963de58..a19498b 100644
--- a/notebooks/unit5/unit5.ipynb
+++ b/notebooks/unit5/unit5.ipynb
@@ -7,7 +7,7 @@
         "colab_type": "text"
       },
       "source": [
-        "<a href=\"https://colab.research.google.com/github/huggingface/deep-rl-class/blob/ThomasSimonini%2FMLAgents/notebooks/unit5/unit5.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+        "<a href=\"https://colab.research.google.com/github/huggingface/deep-rl-class/blob/main/notebooks/unit5/unit5.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
       ]
     },
     {
@@ -218,6 +218,19 @@
         "!pip3 install -e ./ml-agents"
       ]
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "%%capture\n",
+        "# Due to some conflicts between mlagents and tensorflow we need to uninstall it\n",
+        "!pip3 uninstall tensorflow -y"
+      ],
+      "metadata": {
+        "id": "ozcA_tl_XTrv"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "markdown",
       "source": [
@@ -505,6 +518,17 @@
         "id": "KK4fPfnczunT"
       }
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "!mlagents-push-to-hf --run-id=\"SnowballTarget1\" --local-dir=\"./results/SnowballTarget2\" --repo-id=\"ThomasSimonini/ppo-SnowballTargetTESTCOLAB\" --commit-message=\"First Push\""
+      ],
+      "metadata": {
+        "id": "kAFzVB7OYj_H"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -841,4 +865,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}
\ No newline at end of file