diff --git a/01_CARE/exercise.ipynb b/01_CARE/exercise.ipynb
index 5f5d67c..026750e 100644
--- a/01_CARE/exercise.ipynb
+++ b/01_CARE/exercise.ipynb
@@ -647,7 +647,7 @@
    "outputs": [],
    "source": [
     "# Load the model\n",
-    "model = UNet(depth=3, in_channels=1, out_channels=1)"
+    "model = UNet(depth=2, in_channels=1, out_channels=1)"
    ]
   },
   {
@@ -746,10 +746,9 @@
     "![image](nb_data/extensions.png)\n",
     "\n",
     "2) Search Tensorboard and install and install the extension published by Microsoft.\n",
-    "3) Open the command palette (ctrl+shift+p), search for Python: Launch Tensorboard and hit enter.\n",
-    "4) When prompted, select either \"Use current working directory\" or \"Select another folder\" and enter the path to the `01_CARE/runs/` directory.\n",
-    "\n",
-    "Once it's open, continue to the next cell to start training, then open the Tensorboard tab to see the loss curves.\n",
+    "3) Start training. Run the cell below to begin training the model and generating logs.\n",
+    "3) Once training is started. Open the command palette (ctrl+shift+p), search for Python: Launch Tensorboard and hit enter.\n",
+    "4) When prompted, select \"Select another folder\" and enter the path to the `01_CARE/runs/` directory.\n",
     "\n",
     "</div>"
    ]
@@ -772,7 +771,7 @@
    "outputs": [],
    "source": [
     "# Training loop\n",
-    "n_epochs = 10\n",
+    "n_epochs = 5\n",
     "device = \"cuda\" if cuda.is_available() else \"cpu\"\n",
     "model.to(device)\n",
     "\n",
diff --git a/04_DenoiSplit/exercise.ipynb b/04_DenoiSplit/exercise.ipynb
index 76436d0..65b7aab 100644
--- a/04_DenoiSplit/exercise.ipynb
+++ b/04_DenoiSplit/exercise.ipynb
@@ -796,7 +796,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "2c60515d",
+   "id": "37ef9a4b",
    "metadata": {},
    "source": [
     "<hr style=\"height:2px;\"><div class=\"alert alert-block alert-success\"><h1>End of the exercise</h1>\n",