diff --git a/02_Noise2Void/solution.ipynb b/02_Noise2Void/solution.ipynb
index f687e54..684f0be 100755
--- a/02_Noise2Void/solution.ipynb
+++ b/02_Noise2Void/solution.ipynb
@@ -383,26 +383,6 @@
     ]
    },
    "outputs": [],
-   "source": [
-    "# Create a configuration using the helper function\n",
-    "training_config = create_n2v_configuration(\n",
-    "    experiment_name=\"dl4mia_n2v_sem\",\n",
-    "    data_type=\"tiff\",\n",
-    "    axes=\"YX\",\n",
-    "    patch_size=[64, 64],\n",
-    "    batch_size=128,\n",
-    "    num_epochs=10,\n",
-    "    roi_size=3,\n",
-    "    masked_pixel_percentage=0.05,\n",
-    "    logger=\"tensorboard\"\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
    "source": [
     "# Create a configuration using the helper function\n",
     "training_config = create_n2v_configuration(\n",
diff --git a/03_COSDD/solution.ipynb b/03_COSDD/solution.ipynb
index 7aae351..6889d5f 100755
--- a/03_COSDD/solution.ipynb
+++ b/03_COSDD/solution.ipynb
@@ -747,7 +747,7 @@
    "metadata": {},
    "source": [
     "### 2.1. Load test data\n",
-    "The images that we want to denoise are loaded here. These are the same that we used for training, but we'll only load 10 to speed up inference."
+    "The images that we want to denoise are loaded here. These are the same that we used for training, but we'll only load 5 to speed up inference."
    ]
   },
   {
@@ -827,7 +827,11 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "tags": [
+     "task"
+    ]
+   },
    "outputs": [],
    "source": [
     "model_name = ...   ### Insert a string here\n",
@@ -849,7 +853,11 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "tags": [
+     "solution"
+    ]
+   },
    "outputs": [],
    "source": [
     "model_name = \"mito-confocal\"   ### Insert a string here\n",
@@ -983,7 +991,11 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "tags": [
+     "task"
+    ]
+   },
    "outputs": [],
    "source": [
     "use_direct_denoiser = False\n",
@@ -1004,7 +1016,11 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "tags": [
+     "solution"
+    ]
+   },
    "outputs": [],
    "source": [
     "use_direct_denoiser = False\n",