diff --git a/02_production.ipynb b/02_production.ipynb
index 779619c..245f469 100644
--- a/02_production.ipynb
+++ b/02_production.ipynb
@@ -926,8 +926,7 @@
     }
    ],
    "source": [
-    "custom_aug_transforms = [RandomResizedCrop(128, min_scale=0.35), Flip(), Brightness(), Contrast(), Rotate(max_deg=10.0)]\n",
-    "bears = bears.new(item_tfms=Resize(128), batch_tfms=custom_aug_transforms)\n",
+    "bears = bears.new(item_tfms=Resize(128), batch_tfms=aug_transforms(mult=2))\n",
     "dls = bears.dataloaders(path)\n",
     "dls.train.show_batch(max_n=8, nrows=2, unique=True)"
    ]
@@ -2005,34 +2004,9 @@
    "split_at_heading": true
   },
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
-  },
-  "language_info": {
-   "codemirror_mode": {
-    "name": "ipython",
-    "version": 3
-   },
-   "file_extension": ".py",
-   "mimetype": "text/x-python",
-   "name": "python",
-   "nbconvert_exporter": "python",
-   "pygments_lexer": "ipython3",
-   "version": "3.9.13"
-  },
-  "toc": {
-   "base_numbering": 1,
-   "nav_menu": {},
-   "number_sections": true,
-   "sideBar": true,
-   "skip_h1_title": false,
-   "title_cell": "Table of Contents",
-   "title_sidebar": "Contents",
-   "toc_cell": false,
-   "toc_position": {},
-   "toc_section_display": true,
-   "toc_window_display": false
   }
  },
  "nbformat": 4,