diff --git a/how-to-use-azureml/training/train-within-notebook/train-within-notebook.ipynb b/how-to-use-azureml/training/train-within-notebook/train-within-notebook.ipynb
index 95b5df68..085c3dcc 100644
--- a/how-to-use-azureml/training/train-within-notebook/train-within-notebook.ipynb
+++ b/how-to-use-azureml/training/train-within-notebook/train-within-notebook.ipynb
@@ -80,7 +80,8 @@
       "metadata": {
         "tags": [
           "install"
-        ]
+        ],
+        "name": "load_ws"
       },
       "outputs": [],
       "source": [
@@ -155,7 +156,8 @@
         "tags": [
           "local run",
           "outputs upload"
-        ]
+        ],
+        "name": "create_experiment"
       },
       "outputs": [],
       "source": [
@@ -711,4 +713,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 2
-}
\ No newline at end of file
+}