diff --git a/notebooks/chapter_1_introduction/tutorial_0_visualization.ipynb b/notebooks/chapter_1_introduction/tutorial_0_visualization.ipynb
index 7e94a04..aed1212 100644
--- a/notebooks/chapter_1_introduction/tutorial_0_visualization.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_0_visualization.ipynb
@@ -43,18 +43,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_1_grids_and_galaxies.ipynb b/notebooks/chapter_1_introduction/tutorial_1_grids_and_galaxies.ipynb
index 91b95f4..2424de1 100644
--- a/notebooks/chapter_1_introduction/tutorial_1_grids_and_galaxies.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_1_grids_and_galaxies.ipynb
@@ -71,18 +71,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_2_data.ipynb b/notebooks/chapter_1_introduction/tutorial_2_data.ipynb
index 6ff5365..2217381 100644
--- a/notebooks/chapter_1_introduction/tutorial_2_data.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_2_data.ipynb
@@ -57,18 +57,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_3_fitting.ipynb b/notebooks/chapter_1_introduction/tutorial_3_fitting.ipynb
index e99e74a..ca3de09 100644
--- a/notebooks/chapter_1_introduction/tutorial_3_fitting.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_3_fitting.ipynb
@@ -58,18 +58,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_4_methods.ipynb b/notebooks/chapter_1_introduction/tutorial_4_methods.ipynb
index e65fc86..a4626e0 100644
--- a/notebooks/chapter_1_introduction/tutorial_4_methods.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_4_methods.ipynb
@@ -43,18 +43,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_5_summary.ipynb b/notebooks/chapter_1_introduction/tutorial_5_summary.ipynb
index 33a51f8..c77d257 100644
--- a/notebooks/chapter_1_introduction/tutorial_5_summary.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_5_summary.ipynb
@@ -53,18 +53,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_1_non_linear_search.ipynb b/notebooks/chapter_2_modeling/tutorial_1_non_linear_search.ipynb
index 901f9e2..7c04d30 100644
--- a/notebooks/chapter_2_modeling/tutorial_1_non_linear_search.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_1_non_linear_search.ipynb
@@ -190,18 +190,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_2_practicalities.ipynb b/notebooks/chapter_2_modeling/tutorial_2_practicalities.ipynb
index d8f2b1d..9184718 100644
--- a/notebooks/chapter_2_modeling/tutorial_2_practicalities.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_2_practicalities.ipynb
@@ -65,18 +65,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_3_realism_and_complexity.ipynb b/notebooks/chapter_2_modeling/tutorial_3_realism_and_complexity.ipynb
index e0bc470..d317be4 100644
--- a/notebooks/chapter_2_modeling/tutorial_3_realism_and_complexity.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_3_realism_and_complexity.ipynb
@@ -54,18 +54,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_4_dealing_with_failure.ipynb b/notebooks/chapter_2_modeling/tutorial_4_dealing_with_failure.ipynb
index 6a5b67a..fc56c1d 100644
--- a/notebooks/chapter_2_modeling/tutorial_4_dealing_with_failure.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_4_dealing_with_failure.ipynb
@@ -54,18 +54,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_5_linear_profiles.ipynb b/notebooks/chapter_2_modeling/tutorial_5_linear_profiles.ipynb
index be60c29..38894d5 100644
--- a/notebooks/chapter_2_modeling/tutorial_5_linear_profiles.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_5_linear_profiles.ipynb
@@ -68,18 +68,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_6_masking.ipynb b/notebooks/chapter_2_modeling/tutorial_6_masking.ipynb
index 5dfa8ad..f94ce90 100644
--- a/notebooks/chapter_2_modeling/tutorial_6_masking.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_6_masking.ipynb
@@ -43,18 +43,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_7_results.ipynb b/notebooks/chapter_2_modeling/tutorial_7_results.ipynb
index 8ae9c50..75bd00b 100644
--- a/notebooks/chapter_2_modeling/tutorial_7_results.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_7_results.ipynb
@@ -43,18 +43,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb b/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb
index 55a962c..3fac868 100644
--- a/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb
+++ b/notebooks/chapter_2_modeling/tutorial_8_need_for_speed.ipynb
@@ -110,18 +110,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_search_chaining/tutorial_1_search_chaining.ipynb b/notebooks/chapter_3_search_chaining/tutorial_1_search_chaining.ipynb
index 2da6756..67a977c 100644
--- a/notebooks/chapter_3_search_chaining/tutorial_1_search_chaining.ipynb
+++ b/notebooks/chapter_3_search_chaining/tutorial_1_search_chaining.ipynb
@@ -74,18 +74,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_search_chaining/tutorial_2_prior_passing.ipynb b/notebooks/chapter_3_search_chaining/tutorial_2_prior_passing.ipynb
index 22abd14..f81df25 100644
--- a/notebooks/chapter_3_search_chaining/tutorial_2_prior_passing.ipynb
+++ b/notebooks/chapter_3_search_chaining/tutorial_2_prior_passing.ipynb
@@ -51,18 +51,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_search_chaining/tutorial_3_x2_galaxies.ipynb b/notebooks/chapter_3_search_chaining/tutorial_3_x2_galaxies.ipynb
index 640756e..1c3df17 100644
--- a/notebooks/chapter_3_search_chaining/tutorial_3_x2_galaxies.ipynb
+++ b/notebooks/chapter_3_search_chaining/tutorial_3_x2_galaxies.ipynb
@@ -58,18 +58,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_4_pixelizations/tutorial_1_pixelizations.ipynb b/notebooks/chapter_4_pixelizations/tutorial_1_pixelizations.ipynb
index 4928493..95d36e9 100644
--- a/notebooks/chapter_4_pixelizations/tutorial_1_pixelizations.ipynb
+++ b/notebooks/chapter_4_pixelizations/tutorial_1_pixelizations.ipynb
@@ -43,18 +43,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_4_pixelizations/tutorial_2_mappers.ipynb b/notebooks/chapter_4_pixelizations/tutorial_2_mappers.ipynb
index ca3d1f6..cad0c38 100644
--- a/notebooks/chapter_4_pixelizations/tutorial_2_mappers.ipynb
+++ b/notebooks/chapter_4_pixelizations/tutorial_2_mappers.ipynb
@@ -43,18 +43,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_4_pixelizations/tutorial_3_inversions.ipynb b/notebooks/chapter_4_pixelizations/tutorial_3_inversions.ipynb
index 4eb9ae8..904aa73 100644
--- a/notebooks/chapter_4_pixelizations/tutorial_3_inversions.ipynb
+++ b/notebooks/chapter_4_pixelizations/tutorial_3_inversions.ipynb
@@ -47,18 +47,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_4_pixelizations/tutorial_4_bayesian_regularization.ipynb b/notebooks/chapter_4_pixelizations/tutorial_4_bayesian_regularization.ipynb
index fbb889e..8cd2387 100644
--- a/notebooks/chapter_4_pixelizations/tutorial_4_bayesian_regularization.ipynb
+++ b/notebooks/chapter_4_pixelizations/tutorial_4_bayesian_regularization.ipynb
@@ -51,18 +51,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_4_pixelizations/tutorial_5_model_fit.ipynb b/notebooks/chapter_4_pixelizations/tutorial_5_model_fit.ipynb
index 5db0e09..bf0a152 100644
--- a/notebooks/chapter_4_pixelizations/tutorial_5_model_fit.ipynb
+++ b/notebooks/chapter_4_pixelizations/tutorial_5_model_fit.ipynb
@@ -57,18 +57,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/chapter_optional/tutorial_searches.ipynb b/notebooks/chapter_optional/tutorial_searches.ipynb
index d28cb5e..1d7b679 100644
--- a/notebooks/chapter_optional/tutorial_searches.ipynb
+++ b/notebooks/chapter_optional/tutorial_searches.ipynb
@@ -46,18 +46,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/simulators/sersic.ipynb b/notebooks/simulators/sersic.ipynb
index baf8c99..35cb5ad 100644
--- a/notebooks/simulators/sersic.ipynb
+++ b/notebooks/simulators/sersic.ipynb
@@ -48,18 +48,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/simulators/sersic_x2.ipynb b/notebooks/simulators/sersic_x2.ipynb
index e5ccf1b..53521e5 100644
--- a/notebooks/simulators/sersic_x2.ipynb
+++ b/notebooks/simulators/sersic_x2.ipynb
@@ -56,18 +56,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/notebooks/simulators/simple.ipynb b/notebooks/simulators/simple.ipynb
index b5343a9..735f679 100644
--- a/notebooks/simulators/simple.ipynb
+++ b/notebooks/simulators/simple.ipynb
@@ -46,18 +46,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {
diff --git a/start_here.ipynb b/start_here.ipynb
index b40020b..7a2e504 100644
--- a/start_here.ipynb
+++ b/start_here.ipynb
@@ -78,18 +78,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autogalaxy import setup_colab as _setup_colab\n",
+    "else:\n",
+    "    import importlib\n",
     "    import subprocess\n",
     "    import sys\n",
     "\n",
     "    subprocess.check_call(\n",
     "        [sys.executable, \"-m\", \"pip\", \"install\", \"autonerves\", \"--no-deps\"]\n",
     "    )\n",
-    "except ImportError:\n",
-    "    pass\n",
-    "\n",
-    "from autonerves import setup_colab\n",
+    "    _setup_colab = importlib.import_module(\"autonerves.setup_colab\")\n",
     "\n",
-    "setup_colab.setup(\"howtogalaxy\")"
+    "_setup_colab.setup(\"howtogalaxy\")"
    ]
   },
   {