diff --git a/notebooks/bonus-unit1/bonus-unit1.ipynb b/notebooks/bonus-unit1/bonus-unit1.ipynb
index a96ade7..13deda7 100644
--- a/notebooks/bonus-unit1/bonus-unit1.ipynb
+++ b/notebooks/bonus-unit1/bonus-unit1.ipynb
@@ -1,5 +1,15 @@
 {
   "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "view-in-github",
+        "colab_type": "text"
+      },
+      "source": [
+        "<a href=\"https://colab.research.google.com/github/huggingface/deep-rl-class/blob/main/notebooks/bonus-unit1/bonus-unit1.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -320,7 +330,11 @@
       "source": [
         "- **In the case you want to modify the hyperparameters**, in Google Colab notebook, you can click here to open the config.yaml: `/content/ml-agents/config/ppo/Huggy.yaml`\n",
         "\n",
+        "- For instance **if you want to save more models during the training** (for now, we save every 200,000 training timesteps). You need to modify:\n",
+        "  - `checkpoint_interval`: The number of training timesteps collected between each checkpoint.\n",
+        "  - `keep_checkpoints`: The maximum number of model checkpoints to keep. \n",
         "\n",
+        "=> Just keep in mind that **decreasing the `checkpoint_interval` means more models to upload to the Hub and so a longer uploading time** \n",
         "We’re now ready to train our agent 🔥."
       ],
       "metadata": {
@@ -541,7 +555,8 @@
     "accelerator": "GPU",
     "colab": {
       "provenance": [],
-      "private_outputs": true
+      "private_outputs": true,
+      "include_colab_link": true
     },
     "gpuClass": "standard",
     "kernelspec": {