diff --git a/CoRML Demonstration.ipynb b/CoRML Demonstration.ipynb
index 615a9fb..125e2d1 100644
--- a/CoRML Demonstration.ipynb	
+++ b/CoRML Demonstration.ipynb	
@@ -177,6 +177,10 @@
       },
       "outputs": [],
       "source": [
+        "#@title CoRML\n",
+        "#@markdown This cell includes a complete implementation of FPSR model.\n",
+        "#@markdown You can find it at [Github](https://github.com/Joinn99/CoRML/blob/torch/CoRML/model.py).\n",
+        "\n",
         "r\"\"\"\n",
         "CoRML (PyTorch Version)\n",
         "################################################\n",
@@ -527,6 +531,10 @@
         }
       ],
       "source": [
+        "#@title Running a test\n",
+        "#@markdown This cell will run a test on the specified dataset with the model CoRML.\n",
+        "#@markdown Reference [RecBole Doc](https://recbole.io/docs/developer_guide/customize_models.html).\n",
+        "\n",
         "r\"\"\"\n",
         "Code reference: https://recbole.io/docs/developer_guide/customize_models.html\n",
         "\"\"\"\n",