diff --git a/data_prep.ipynb b/data_prep.ipynb
index 98185d9..64f44dc 100644
--- a/data_prep.ipynb
+++ b/data_prep.ipynb
@@ -1422,7 +1422,7 @@
     "    window_bound: tuple,  # bounding box in format (minx, miny, maxx, maxy)\n",
     "    array: np.ndarray,  # must be in CHW format (channel, height, width)\n",
     "    save_netcdf: bool = False,  # whether to also save a NetCDF file\n",
-    "    crs: str = \"EPSG:3031\",  # projected coordinate system to use\n",
+    "    crs: str = \"+proj=stere +lat_0=-90 +lat_ts=-71 +lon_0=0 +k=1 +x_0=0 +y_0=0 +datum=WGS84 +units=m +no_defs\",  # projected coordinate system to use\n",
     "    dtype: str = None,  # data type to use e.g. np.float32, default inferred from array\n",
     "    nodataval: float = -2000,  # what to use as NaN, hardcoded default to -2000m\n",
     "    tiled: bool = False,  # store data arranged in square tiles, default is False\n",
diff --git a/data_prep.py b/data_prep.py
index 7916c00..c90158d 100644
--- a/data_prep.py
+++ b/data_prep.py
@@ -766,7 +766,7 @@ def save_array_to_grid(
     window_bound: tuple,  # bounding box in format (minx, miny, maxx, maxy)
     array: np.ndarray,  # must be in CHW format (channel, height, width)
     save_netcdf: bool = False,  # whether to also save a NetCDF file
-    crs: str = "EPSG:3031",  # projected coordinate system to use
+    crs: str = "+proj=stere +lat_0=-90 +lat_ts=-71 +lon_0=0 +k=1 +x_0=0 +y_0=0 +datum=WGS84 +units=m +no_defs",  # projected coordinate system to use
     dtype: str = None,  # data type to use e.g. np.float32, default inferred from array
     nodataval: float = -2000,  # what to use as NaN, hardcoded default to -2000m
     tiled: bool = False,  # store data arranged in square tiles, default is False
diff --git a/deepbedmap.ipynb b/deepbedmap.ipynb
index 32fd653..f6ecc35 100644
--- a/deepbedmap.ipynb
+++ b/deepbedmap.ipynb
@@ -671,28 +671,28 @@
     "subplot(directive=\"begin\", row=2, col=2, A=\"+jCT+o-4c/-5c\", Fs=\"9c/9c\", M=\"2c/3c\")\n",
     "plot_3d_view(\n",
     "    img=\"model/deepbedmap3.nc\",  # DeepBedMap\n",
-    "    ax=(1, 1),\n",
+    "    ax=(0, 0),\n",
     "    zmin=-1400,\n",
     "    title=\"DeepBedMap\",  # ours\n",
     "    zlabel=\"Bed elevation (metres)\",\n",
     ")\n",
     "plot_3d_view(\n",
     "    img=\"model/cubicbedmap.nc\",  # BEDMAP2\n",
-    "    ax=(1, 2),\n",
+    "    ax=(0, 1),\n",
     "    zmin=-1400,\n",
     "    title=\"BEDMAP2\",\n",
     "    zlabel=\"Bed elevation (metres)\",\n",
     ")\n",
     "plot_3d_view(\n",
     "    img=\"model/elevdiffmap.nc\",  # DeepBedMap - BEDMAP2\n",
-    "    ax=(2, 1),\n",
+    "    ax=(1, 0),\n",
     "    zmin=-400,\n",
     "    title=\"Elevation Diff\",\n",
     "    zlabel=\"Difference (metres)\",\n",
     ")\n",
     "plot_3d_view(\n",
     "    img=\"model/synthetichr.nc\",  # Synthetic High Resolution product\n",
-    "    ax=(2, 2),\n",
+    "    ax=(1, 1),\n",
     "    zmin=-1400,\n",
     "    title=\"Synthetic HRES\",\n",
     "    zlabel=\"Bed elevation (metres)\",\n",
diff --git a/deepbedmap.py b/deepbedmap.py
index 0a95c8e..f4445ca 100644
--- a/deepbedmap.py
+++ b/deepbedmap.py
@@ -452,28 +452,28 @@ def load_trained_model(
 subplot(directive="begin", row=2, col=2, A="+jCT+o-4c/-5c", Fs="9c/9c", M="2c/3c")
 plot_3d_view(
     img="model/deepbedmap3.nc",  # DeepBedMap
-    ax=(1, 1),
+    ax=(0, 0),
     zmin=-1400,
     title="DeepBedMap",  # ours
     zlabel="Bed elevation (metres)",
 )
 plot_3d_view(
     img="model/cubicbedmap.nc",  # BEDMAP2
-    ax=(1, 2),
+    ax=(0, 1),
     zmin=-1400,
     title="BEDMAP2",
     zlabel="Bed elevation (metres)",
 )
 plot_3d_view(
     img="model/elevdiffmap.nc",  # DeepBedMap - BEDMAP2
-    ax=(2, 1),
+    ax=(1, 0),
     zmin=-400,
     title="Elevation Diff",
     zlabel="Difference (metres)",
 )
 plot_3d_view(
     img="model/synthetichr.nc",  # Synthetic High Resolution product
-    ax=(2, 2),
+    ax=(1, 1),
     zmin=-1400,
     title="Synthetic HRES",
     zlabel="Bed elevation (metres)",