diff --git a/exercise.ipynb b/exercise.ipynb
index 259bee1..d880853 100644
--- a/exercise.ipynb
+++ b/exercise.ipynb
@@ -567,9 +567,7 @@
   {
    "cell_type": "markdown",
    "id": "0d2923bc",
-   "metadata": {
-    "lines_to_next_cell": 0
-   },
+   "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\">\n",
     "<h2> Checkpoint 1 </h2>\n",
@@ -736,10 +734,10 @@
     "threshold = ...\n",
     "\n",
     "# Get inner mask\n",
-    "inner_mask = get_inner_mask(pred, threshold=threshold)\n",
+    "inner_mask = get_inner_mask(...)\n",
     "\n",
     "# Get the segmentation\n",
-    "seg = watershed_from_boundary_distance(pred, inner_mask, min_seed_distance=20)"
+    "seg = watershed_from_boundary_distance(...)"
    ]
   },
   {
@@ -1366,9 +1364,7 @@
   {
    "cell_type": "markdown",
    "id": "8ad1901f",
-   "metadata": {
-    "lines_to_next_cell": 0
-   },
+   "metadata": {},
    "source": [
     "<hr style=\"height:2px;\">\n",
     "\n",
@@ -1376,6 +1372,15 @@
     "[Here](https://localshapedescriptors.github.io/) is a blog post describing the Local Shape Descriptor method of instance segmentation.\n"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "44f53870",
+   "metadata": {},
+   "source": [
+    "<div class=\"alert alert-block alert-success\">\n",
+    "<h2> Checkpoint 3 </h2>"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "5c861567",
diff --git a/solution.ipynb b/solution.ipynb
index abe2c9e..ab2898a 100644
--- a/solution.ipynb
+++ b/solution.ipynb
@@ -602,9 +602,7 @@
   {
    "cell_type": "markdown",
    "id": "756a7002",
-   "metadata": {
-    "lines_to_next_cell": 0
-   },
+   "metadata": {},
    "source": [
     "<div class=\"alert alert-block alert-success\">\n",
     "<h2> Checkpoint 1 </h2>\n",
@@ -1421,9 +1419,7 @@
   {
    "cell_type": "markdown",
    "id": "c78560d7",
-   "metadata": {
-    "lines_to_next_cell": 0
-   },
+   "metadata": {},
    "source": [
     "<hr style=\"height:2px;\">\n",
     "\n",
@@ -1431,6 +1427,15 @@
     "[Here](https://localshapedescriptors.github.io/) is a blog post describing the Local Shape Descriptor method of instance segmentation.\n"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "8b3fd165",
+   "metadata": {},
+   "source": [
+    "<div class=\"alert alert-block alert-success\">\n",
+    "<h2> Checkpoint 3 </h2>"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "64c2fb7f",