diff --git a/notebooks/chapter_1_introduction/start_here.ipynb b/notebooks/chapter_1_introduction/start_here.ipynb
index 6efd7aa..a00630c 100644
--- a/notebooks/chapter_1_introduction/start_here.ipynb
+++ b/notebooks/chapter_1_introduction/start_here.ipynb
@@ -82,18 +82,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_1_models.ipynb b/notebooks/chapter_1_introduction/tutorial_1_models.ipynb
index ee18144..f46031b 100644
--- a/notebooks/chapter_1_introduction/tutorial_1_models.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_1_models.ipynb
@@ -97,18 +97,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_2_fitting_data.ipynb b/notebooks/chapter_1_introduction/tutorial_2_fitting_data.ipynb
index 58d759c..1aab95d 100644
--- a/notebooks/chapter_1_introduction/tutorial_2_fitting_data.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_2_fitting_data.ipynb
@@ -81,18 +81,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_3_non_linear_search.ipynb b/notebooks/chapter_1_introduction/tutorial_3_non_linear_search.ipynb
index 6152757..5fcb9ea 100644
--- a/notebooks/chapter_1_introduction/tutorial_3_non_linear_search.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_3_non_linear_search.ipynb
@@ -173,18 +173,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_4_why_modeling_is_hard.ipynb b/notebooks/chapter_1_introduction/tutorial_4_why_modeling_is_hard.ipynb
index 8a2dc26..1e808cf 100644
--- a/notebooks/chapter_1_introduction/tutorial_4_why_modeling_is_hard.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_4_why_modeling_is_hard.ipynb
@@ -70,18 +70,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_1_introduction/tutorial_5_results_and_samples.ipynb b/notebooks/chapter_1_introduction/tutorial_5_results_and_samples.ipynb
index 9c8defb..733958f 100644
--- a/notebooks/chapter_1_introduction/tutorial_5_results_and_samples.ipynb
+++ b/notebooks/chapter_1_introduction/tutorial_5_results_and_samples.ipynb
@@ -62,18 +62,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_graphical_models/tutorial_1_individual_models.ipynb b/notebooks/chapter_3_graphical_models/tutorial_1_individual_models.ipynb
index 169158a..1b78c68 100644
--- a/notebooks/chapter_3_graphical_models/tutorial_1_individual_models.ipynb
+++ b/notebooks/chapter_3_graphical_models/tutorial_1_individual_models.ipynb
@@ -79,18 +79,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_graphical_models/tutorial_2_graphical_model.ipynb b/notebooks/chapter_3_graphical_models/tutorial_2_graphical_model.ipynb
index 1fa63c1..2240c10 100644
--- a/notebooks/chapter_3_graphical_models/tutorial_2_graphical_model.ipynb
+++ b/notebooks/chapter_3_graphical_models/tutorial_2_graphical_model.ipynb
@@ -63,18 +63,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_graphical_models/tutorial_3_graphical_benefits.ipynb b/notebooks/chapter_3_graphical_models/tutorial_3_graphical_benefits.ipynb
index ae32fae..fe32625 100644
--- a/notebooks/chapter_3_graphical_models/tutorial_3_graphical_benefits.ipynb
+++ b/notebooks/chapter_3_graphical_models/tutorial_3_graphical_benefits.ipynb
@@ -66,18 +66,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_graphical_models/tutorial_4_hierachical_models.ipynb b/notebooks/chapter_3_graphical_models/tutorial_4_hierachical_models.ipynb
index f471626..9602640 100644
--- a/notebooks/chapter_3_graphical_models/tutorial_4_hierachical_models.ipynb
+++ b/notebooks/chapter_3_graphical_models/tutorial_4_hierachical_models.ipynb
@@ -63,18 +63,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_graphical_models/tutorial_5_expectation_propagation.ipynb b/notebooks/chapter_3_graphical_models/tutorial_5_expectation_propagation.ipynb
index d13b94d..9aaf16e 100644
--- a/notebooks/chapter_3_graphical_models/tutorial_5_expectation_propagation.ipynb
+++ b/notebooks/chapter_3_graphical_models/tutorial_5_expectation_propagation.ipynb
@@ -63,18 +63,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_ep.ipynb b/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_ep.ipynb
index 08cd26d..9bb7081 100644
--- a/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_ep.ipynb
+++ b/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_ep.ipynb
@@ -48,18 +48,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_individual.ipynb b/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_individual.ipynb
index dada37d..3fb42cb 100644
--- a/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_individual.ipynb
+++ b/notebooks/chapter_3_graphical_models/tutorial_optional_hierarchical_individual.ipynb
@@ -53,18 +53,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/simulators/simulators.ipynb b/notebooks/simulators/simulators.ipynb
index eca37e0..c6ced6c 100644
--- a/notebooks/simulators/simulators.ipynb
+++ b/notebooks/simulators/simulators.ipynb
@@ -54,18 +54,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/simulators/simulators_sample.ipynb b/notebooks/simulators/simulators_sample.ipynb
index 970af56..384c5b7 100644
--- a/notebooks/simulators/simulators_sample.ipynb
+++ b/notebooks/simulators/simulators_sample.ipynb
@@ -41,18 +41,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/notebooks/simulators/util.ipynb b/notebooks/simulators/util.ipynb
index 45fe4b5..5db1bbe 100644
--- a/notebooks/simulators/util.ipynb
+++ b/notebooks/simulators/util.ipynb
@@ -23,18 +23,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {
diff --git a/start_here.ipynb b/start_here.ipynb
index 9d092e9..54a3356 100644
--- a/start_here.ipynb
+++ b/start_here.ipynb
@@ -77,18 +77,19 @@
    "source": [
     "try:\n",
     "    import google.colab\n",
+    "except ImportError:\n",
+    "    from autofit 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(\"howtofit\")"
+    "_setup_colab.setup(\"howtofit\")"
    ]
   },
   {