diff --git a/nanshe_ipython.ipynb b/nanshe_ipython.ipynb
index ac282ac..e11435d 100644
--- a/nanshe_ipython.ipynb
+++ b/nanshe_ipython.ipynb
@@ -378,6 +378,10 @@
       "source": [
         "front = 0\n",
         "back = 0\n",
+        "top = 0\n",
+        "bottom = 0\n",
+        "left = 0\n",
+        "right = 0\n",
         "\n",
         "\n",
         "for k in [subgroup_trim]:\n",
@@ -389,7 +393,7 @@
         "da_imgs = dask_store[subgroup_raw]\n",
         "\n",
         "# Trim frames from front and back\n",
-        "da_imgs_trim = da_imgs[front:len(da_imgs)-back]\n",
+        "da_imgs_trim = da_imgs[front:da_imgs.shape[0]-back, top:da_imgs.shape[1]-bottom, left:da_imgs.shape[2]-right]\n",
         "\n",
         "# Store trimmed data\n",
         "dask_store[subgroup_trim] = da_imgs_trim\n",