diff --git a/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb b/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb
index 81845de..7d5ab53 100644
--- a/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb
+++ b/notebooks/06_end_to_end/slice_based/01_fracatlas_classification.ipynb
@@ -54,14 +54,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 5,
    "id": "734f7a66",
    "metadata": {},
    "outputs": [],
    "source": [
     "from datamint import Api\n",
     "\n",
-    "PROJECT_NAME = \"FracAtlas\"\n",
+    "PROJECT_NAME = \"ClassificationTest\"\n",
     "api = Api()"
    ]
   },
@@ -106,10 +106,29 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 8,
    "id": "bb794bc8",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "INFO:datamint.dataset.factory:Detected resource type 'image'; using ImageDataset.\n",
+      "INFO:datamint.dataset.base:Allowing external image label '('has_fracture', 'no')' not in project specs.\n",
+      "INFO:datamint.dataset.base:Allowing external image label '('has_fracture', 'yes')' not in project specs.\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "train : 3268 resources\n",
+      "val   :  408 resources\n",
+      "test  :  409 resources\n"
+     ]
+    }
+   ],
    "source": [
     "from datamint.dataset import build_dataset\n",
     "\n",
@@ -121,7 +140,7 @@
     ")\n",
     "\n",
     "parts = dataset.split(train=0.8, val=0.1, test=0.1, seed=123)\n",
-    "parts.save()  \n",
+    "parts.save(force=True)  \n",
     "\n",
     "for name, ds in parts.items():\n",
     "    print(f'{name:6s}: {len(ds):4d} resources')"
@@ -165,8 +184,9 @@
     "    project=PROJECT_NAME,\n",
     "    image_size=384,\n",
     "    batch_size=8,\n",
-    "    max_epochs=10,\n",
+    "    max_epochs=3,\n",
     "    accelerator='auto',\n",
+    "    pretrained=False,\n",
     "    # model_name='MyModelName'  # defaults to project name\n",
     ")\n",
     "\n",