diff --git a/1_getting_started/1a_train_keras.ipynb b/1_getting_started/1a_train_keras.ipynb
index 49c8e9ef..58ed86f3 100644
--- a/1_getting_started/1a_train_keras.ipynb
+++ b/1_getting_started/1a_train_keras.ipynb
@@ -178,6 +178,24 @@
     "model.save('../models/keras_model_part1.h5')"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "d2b36d68",
+   "metadata": {},
+   "source": [
+    "## Copy plotting module\n",
+    "Copy the plotting module before checking the performance:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "cp ../plotting.py ./"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},