diff --git a/Test_S1_Mosaic.ipynb b/Test_S1_Mosaic.ipynb
index d9b3dea..0b4d512 100644
--- a/Test_S1_Mosaic.ipynb
+++ b/Test_S1_Mosaic.ipynb
@@ -101,9 +101,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Do the processing, all in memory\n",
-    "loaded = output_data.compute()\n",
-    "loaded"
+    "output_data[\"count\"].plot.imshow(size=10)"
    ]
   },
   {
@@ -112,7 +110,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "loaded[[\"median_vv\", \"median_vh\", \"median_vv_vh\"]].to_array().plot.imshow(size=10, vmin=0, vmax=0.1)"
+    "output_data[[\"median_vv\", \"median_vh\", \"median_vv_vh\"]].to_array().plot.imshow(size=10, vmin=0, vmax=0.1)"
    ]
   },
   {
@@ -121,7 +119,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "loaded[[\"mean_vv\", \"mean_vh\", \"mean_vv_vh\"]].to_array().plot.imshow(size=10, vmin=0, vmax=0.2)"
+    "output_data[[\"mean_vv\", \"mean_vh\", \"mean_vv_vh\"]].to_array().plot.imshow(size=10, vmin=0, vmax=0.2)"
    ]
   },
   {