diff --git a/argilla/docs/tutorials/image_classification.ipynb b/argilla/docs/tutorials/image_classification.ipynb
index 5d01954eb5..4202e2d86e 100644
--- a/argilla/docs/tutorials/image_classification.ipynb
+++ b/argilla/docs/tutorials/image_classification.ipynb
@@ -475,6 +475,13 @@
     "    Check this [how-to guide](../how_to_guides/query.md) to know more about filtering and querying in Argilla. Also, you can check the Hugging Face docs on [fine-tuning an image classification model](https://huggingface.co/docs/transformers/en/tasks/image_classification)."
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Formatting the data"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 16,
@@ -608,6 +615,13 @@
     "prepared_ds"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### The actual training"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},