diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index a10cea1cb..33a4067aa 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,7 +1,7 @@
 ---
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v3.4.0
+    rev: v4.1.0
     hooks:
       - id: trailing-whitespace
         exclude: ^.*fits
@@ -14,18 +14,18 @@ repos:
       - id: end-of-file-fixer
         exclude: ^.*fits
   - repo: https://github.com/psf/black
-    rev: 20.8b1
+    rev: 22.1.0
     hooks:
       - id: black
         language_version: python3
         args:
           - "--line-length=100"
   - repo: https://github.com/asottile/reorder_python_imports
-    rev: v2.4.0
+    rev: v2.7.1
     hooks:
       - id: reorder-python-imports
   - repo: https://github.com/pycqa/flake8
-    rev: 3.8.4
+    rev: 4.0.1
     hooks:
       - id: flake8
         args: # arguments to configure flake8
@@ -35,7 +35,7 @@ repos:
           - "--ignore=E203, W503"
           - "--exclude=btk/__init__.py"
   - repo: https://github.com/pycqa/pydocstyle
-    rev: 6.0.0 # pick a git hash / tag to point to
+    rev: 6.1.1 # pick a git hash / tag to point to
     hooks:
       - id: pydocstyle
         args:
@@ -46,3 +46,9 @@ repos:
               ^conftest.py|
               ^tests/
           )
+  - repo: https://github.com/kynan/nbstripout
+    rev: 0.5.0
+    hooks:
+      - id: nbstripout
+        args:
+          - --keep-output
diff --git a/btk/catalog.py b/btk/catalog.py
index 21f357d9d..fb2fa523b 100644
--- a/btk/catalog.py
+++ b/btk/catalog.py
@@ -90,7 +90,7 @@ def _prepare_table(self, raw_catalog):
         )
         r_sec = np.hypot(
             raw_catalog["a_d"] * (1 - f) ** 0.5 * 4.66,
-            raw_catalog["a_b"] * f ** 0.5 * 1.46,
+            raw_catalog["a_b"] * f**0.5 * 1.46,
         )
         # BTK now requires ref_mags, but Catsim still wants magnitudes
         table["ref_mag"] = raw_catalog["i_ab"]
diff --git a/btk/sampling_functions.py b/btk/sampling_functions.py
index f926f4bc7..93e25f2d2 100644
--- a/btk/sampling_functions.py
+++ b/btk/sampling_functions.py
@@ -188,7 +188,7 @@ def __call__(self, table, **kwargs):
         blend_table["ra"] = 0.0
         blend_table["dec"] = 0.0
         # keep number density of objects constant
-        maxshift = self.stamp_size / 30.0 * number_of_objects ** 0.5
+        maxshift = self.stamp_size / 30.0 * number_of_objects**0.5
         x_peak, y_peak = _get_random_center_shift(number_of_objects + 1, maxshift, self.rng)
         blend_table["ra"] += x_peak
         blend_table["dec"] += y_peak
diff --git a/btk/survey.py b/btk/survey.py
index a6e4e07ca..ac42881f0 100644
--- a/btk/survey.py
+++ b/btk/survey.py
@@ -250,7 +250,7 @@ def get_mean_sky_level(survey, filt):
     Returns:
         Corresponding mean sky level
     """
-    return get_flux(filt.sky_brightness, filt, survey) * survey.pixel_scale ** 2
+    return get_flux(filt.sky_brightness, filt, survey) * survey.pixel_scale**2
 
 
 def make_wcs(pixel_scale, shape, center_pix=None, center_sky=None, projection="TAN"):
diff --git a/notebooks/00-intro.ipynb b/notebooks/00-intro.ipynb
index 2a0d3fc0c..1f027d1b0 100644
--- a/notebooks/00-intro.ipynb
+++ b/notebooks/00-intro.ipynb
@@ -11,13 +11,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 25,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:45.666746Z",
-     "start_time": "2022-02-18T13:36:45.633723Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "%matplotlib inline\n",
@@ -29,13 +24,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 26,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:45.751253Z",
-     "start_time": "2022-02-18T13:36:45.723674Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "import btk\n",
@@ -63,13 +53,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 27,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:46.367353Z",
-     "start_time": "2022-02-18T13:36:46.340630Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "from btk import DEFAULT_SEED\n",
@@ -78,12 +63,7 @@
   },
   {
    "cell_type": "markdown",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-10-28T21:09:53.185299Z",
-     "start_time": "2021-10-28T21:09:52.872533Z"
-    }
-   },
+   "metadata": {},
    "source": [
     "Every object in BTK that needs a seed uses `DEFAULT_SEED` implicitly. In this tutorial we explicitly show how this seed is passed in and which objects needs a seed. However, we omit it in the other tutorials."
    ]
@@ -108,13 +88,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 28,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:47.776667Z",
-     "start_time": "2022-02-18T13:36:47.730825Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "catalog_name = \"../data/sample_input_catalog.fits\"\n",
@@ -130,13 +105,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 29,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:48.305841Z",
-     "start_time": "2022-02-18T13:36:48.260740Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "_, ext = os.path.splitext(catalog_name)\n",
@@ -163,13 +133,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 30,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:50.632300Z",
-     "start_time": "2022-02-18T13:36:50.604846Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "stamp_size = 24.0  # Size of the stamp, in arcseconds\n",
@@ -189,13 +154,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 31,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:50.976724Z",
-     "start_time": "2022-02-18T13:36:50.945358Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "\n",
@@ -273,13 +233,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 32,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:52.148061Z",
-     "start_time": "2022-02-18T13:36:51.973115Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "Rubin = btk.survey.get_surveys(\"Rubin\")"
@@ -294,13 +249,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 33,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:52.699609Z",
-     "start_time": "2022-02-18T13:36:52.670264Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "from btk.survey import get_psf\n",
@@ -413,13 +363,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 34,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:53.098476Z",
-     "start_time": "2022-02-18T13:36:53.076470Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "def random_psf():\n",
@@ -459,13 +404,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 35,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:36:54.444423Z",
-     "start_time": "2022-02-18T13:36:54.416901Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "draw_generator = btk.draw_blends.CatsimGenerator(\n",
@@ -496,14 +436,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 38,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:33.462443Z",
-     "start_time": "2022-02-18T13:37:29.022131Z"
-    },
-    "scrolled": true
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -596,13 +530,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 39,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:33.955864Z",
-     "start_time": "2022-02-18T13:37:33.612078Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -633,13 +562,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 40,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:34.406252Z",
-     "start_time": "2022-02-18T13:37:34.376889Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "def sep_measure(batch, idx, channels_last=False, surveys=None, sigma_noise=1.5, **kwargs):\n",
@@ -718,13 +642,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 41,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:35.098713Z",
-     "start_time": "2022-02-18T13:37:35.075138Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [],
    "source": [
     "meas_generator = btk.measure.MeasureGenerator(btk.measure.sep_measure,draw_generator)"
@@ -739,13 +658,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 42,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:39.265995Z",
-     "start_time": "2022-02-18T13:37:35.513639Z"
-    }
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -789,14 +703,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 43,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:45.379185Z",
-     "start_time": "2022-02-18T13:37:39.268284Z"
-    },
-    "scrolled": true
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -834,13 +742,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 44,
+   "execution_count": null,
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:45.464146Z",
-     "start_time": "2022-02-18T13:37:45.380838Z"
-    },
-    "scrolled": false,
     "tags": [
      "nbval-ignore-output"
     ]
@@ -874,14 +777,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 45,
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:48.074747Z",
-     "start_time": "2022-02-18T13:37:45.465252Z"
-    },
-    "scrolled": false
-   },
+   "execution_count": null,
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -970,12 +867,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 46,
+   "execution_count": null,
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:48.098310Z",
-     "start_time": "2022-02-18T13:37:48.076910Z"
-    },
     "tags": [
      "nbval-skip"
     ]
@@ -1004,12 +897,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 47,
+   "execution_count": null,
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:54.869668Z",
-     "start_time": "2022-02-18T13:37:48.099758Z"
-    },
     "tags": [
      "nbval-skip"
     ]
@@ -1051,12 +940,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 48,
+   "execution_count": null,
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:37:55.330081Z",
-     "start_time": "2022-02-18T13:37:54.871227Z"
-    },
     "tags": [
      "nbval-skip"
     ]
diff --git a/notebooks/01a-cosmos_tutorial.ipynb b/notebooks/01a-cosmos_tutorial.ipynb
index fe8dc1993..0447afdf6 100644
--- a/notebooks/01a-cosmos_tutorial.ipynb
+++ b/notebooks/01a-cosmos_tutorial.ipynb
@@ -12,14 +12,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "id": "4eefbe0a",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:33.122916Z",
-     "start_time": "2022-02-18T13:38:31.547894Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "%matplotlib inline\n",
@@ -56,14 +51,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "id": "0b49f336",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:33.339539Z",
-     "start_time": "2022-02-18T13:38:33.124308Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "stamp_size = 24.0\n",
@@ -85,14 +75,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "id": "de364e04",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:33.786332Z",
-     "start_time": "2022-02-18T13:38:33.780912Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "gal_type = \"real\" # to choose between real and parametric models\n",
@@ -111,15 +96,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "id": "e7bb0a6e",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:36.276414Z",
-     "start_time": "2022-02-18T13:38:34.214429Z"
-    },
-    "scrolled": false
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
diff --git a/notebooks/01b-scarlet-measure.ipynb b/notebooks/01b-scarlet-measure.ipynb
index 3d6054c81..958aebdb8 100644
--- a/notebooks/01b-scarlet-measure.ipynb
+++ b/notebooks/01b-scarlet-measure.ipynb
@@ -2,14 +2,9 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "id": "248f413f",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:25:06.132842Z",
-     "start_time": "2021-07-16T21:25:04.789315Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "%load_ext autoreload\n",
@@ -39,14 +34,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "id": "454ca993",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:25:06.421265Z",
-     "start_time": "2021-07-16T21:25:06.134720Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "catalog_name = \"../data/sample_input_catalog.fits\"\n",
@@ -56,14 +46,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "id": "ae95cda3",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:25:06.454142Z",
-     "start_time": "2021-07-16T21:25:06.422695Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "catalog = btk.catalog.CatsimCatalog.from_file(catalog_name)\n",
@@ -78,14 +63,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "id": "41693b09",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:25:06.491386Z",
-     "start_time": "2021-07-16T21:25:06.456424Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "def scarlet_measure(batch,idx,channels_last=False, is_multiresolution=False,**kwargs):\n",
@@ -217,14 +197,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "id": "5cc9d875",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:25:06.511153Z",
-     "start_time": "2021-07-16T21:25:06.493100Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "measure_kwargs=[{\"sigma_noise\": 2.0}]\n",
@@ -240,14 +215,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "id": "3e363a89",
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:26:43.809035Z",
-     "start_time": "2021-07-16T21:25:06.512545Z"
-    },
-    "scrolled": true,
     "tags": [
      "nbval-ignore-output"
     ]
@@ -277,14 +247,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "id": "aee3c7f5",
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:27:35.572287Z",
-     "start_time": "2021-07-16T21:27:35.389463Z"
-    },
-    "scrolled": false,
     "tags": [
      "nbval-ignore-output"
     ]
@@ -311,15 +276,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "id": "8dd1fc73",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2021-07-16T21:28:07.429050Z",
-     "start_time": "2021-07-16T21:28:02.700710Z"
-    },
-    "scrolled": false
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
diff --git a/notebooks/02a-multi-tutorial.ipynb b/notebooks/02a-multi-tutorial.ipynb
index 90792ea20..3fe1d46aa 100644
--- a/notebooks/02a-multi-tutorial.ipynb
+++ b/notebooks/02a-multi-tutorial.ipynb
@@ -2,14 +2,9 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "id": "b8bc0992",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:45.445007Z",
-     "start_time": "2022-02-18T13:38:45.231507Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "# Autoreload packages in case they change.\n",
@@ -64,14 +59,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "id": "8d4d179f",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:46.790725Z",
-     "start_time": "2022-02-18T13:38:46.622302Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "sampling_function = btk.sampling_functions.DefaultSampling()\n",
@@ -98,14 +88,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "id": "a1605828",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:48.881069Z",
-     "start_time": "2022-02-18T13:38:47.706342Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -205,14 +190,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "id": "dbfd0584",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:50.175990Z",
-     "start_time": "2022-02-18T13:38:49.866207Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -257,14 +237,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": null,
    "id": "aaa0fcbf",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:51.651476Z",
-     "start_time": "2022-02-18T13:38:51.146380Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -305,14 +280,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": null,
    "id": "0066f9c9",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:52.829085Z",
-     "start_time": "2022-02-18T13:38:52.801669Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "name": "stdout",
@@ -369,13 +339,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": null,
    "id": "6374c0e0",
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:55.811717Z",
-     "start_time": "2022-02-18T13:38:55.677482Z"
-    },
     "tags": [
      "nbval-ignore-output"
     ]
@@ -413,14 +379,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": null,
    "id": "875bed15",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:57.946427Z",
-     "start_time": "2022-02-18T13:38:57.763393Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "sampling_function = btk.sampling_functions.DefaultSampling()\n",
@@ -441,14 +402,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": null,
    "id": "1d213d03",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:58.843704Z",
-     "start_time": "2022-02-18T13:38:58.441118Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -471,14 +427,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": null,
    "id": "cda15f69",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:38:59.549831Z",
-     "start_time": "2022-02-18T13:38:58.845675Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "name": "stdout",
@@ -543,14 +494,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": null,
    "id": "55243a7e",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:01.053896Z",
-     "start_time": "2022-02-18T13:39:00.908695Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "sampling_function = btk.sampling_functions.DefaultSampling()\n",
@@ -576,14 +522,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": null,
    "id": "8b151e56",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:01.815683Z",
-     "start_time": "2022-02-18T13:39:01.359465Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -606,14 +547,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 20,
+   "execution_count": null,
    "id": "9fb29875",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:02.556500Z",
-     "start_time": "2022-02-18T13:39:01.957419Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "name": "stdout",
@@ -677,14 +613,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": null,
    "id": "1edac435",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:04.618045Z",
-     "start_time": "2022-02-18T13:39:03.400041Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
diff --git a/notebooks/02b-custom-tutorial.ipynb b/notebooks/02b-custom-tutorial.ipynb
index e27c58170..40fbc164d 100644
--- a/notebooks/02b-custom-tutorial.ipynb
+++ b/notebooks/02b-custom-tutorial.ipynb
@@ -2,14 +2,9 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "id": "b8bc0992",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:22.581962Z",
-     "start_time": "2022-02-18T13:39:20.888108Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "# Autoreload packages in case they change.\n",
@@ -56,14 +51,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "id": "210daf61",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:22.605085Z",
-     "start_time": "2022-02-18T13:39:22.584213Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "class DefaultSampling(btk.sampling_functions.SamplingFunction):\n",
@@ -149,14 +139,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "id": "be9b1a2b",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:22.624901Z",
-     "start_time": "2022-02-18T13:39:22.607230Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "class PairSampling(btk.sampling_functions.SamplingFunction):\n",
@@ -218,14 +203,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "id": "f1c8faa5",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:25.986910Z",
-     "start_time": "2022-02-18T13:39:25.710638Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "sampling_function = PairSampling()\n",
@@ -244,14 +224,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "id": "b067b8c9",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:27.155387Z",
-     "start_time": "2022-02-18T13:39:26.078286Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -380,14 +355,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "id": "889b90b8",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:28.213249Z",
-     "start_time": "2022-02-18T13:39:28.180846Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "from btk.survey import Survey, Filter, get_psf\n",
@@ -492,14 +462,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "id": "ca76bc80",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:28.465231Z",
-     "start_time": "2022-02-18T13:39:28.427684Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "sampling_function = btk.sampling_functions.DefaultSampling()\n",
@@ -518,14 +483,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "id": "ae6d8c48",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:29.673383Z",
-     "start_time": "2022-02-18T13:39:28.836219Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -628,14 +588,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "id": "9a782feb",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:29.742788Z",
-     "start_time": "2022-02-18T13:39:29.711632Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import sep\n",
@@ -741,14 +696,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "id": "5ef91dac",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:31.195227Z",
-     "start_time": "2022-02-18T13:39:31.040530Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "catalog_name = \"../data/sample_input_catalog.fits\"\n",
@@ -775,14 +725,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "id": "786178c6",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:31.786772Z",
-     "start_time": "2022-02-18T13:39:31.756495Z"
-    }
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "def meas_ksb_ellipticity(image, additional_params):\n",
@@ -817,14 +762,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": null,
    "id": "58ae7ce9",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:37.539900Z",
-     "start_time": "2022-02-18T13:39:32.112238Z"
-    }
-   },
+   "metadata": {},
    "outputs": [
     {
      "data": {
@@ -858,13 +798,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": null,
    "id": "b88cce66",
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:37.569957Z",
-     "start_time": "2022-02-18T13:39:37.543843Z"
-    },
     "tags": [
      "nbval-ignore-output"
     ]
@@ -886,13 +822,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": null,
    "id": "177245eb",
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:39.818997Z",
-     "start_time": "2022-02-18T13:39:39.790947Z"
-    },
     "tags": [
      "nbval-ignore-output"
     ]
@@ -926,14 +858,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": null,
    "id": "3e13d238",
    "metadata": {
-    "ExecuteTime": {
-     "end_time": "2022-02-18T13:39:40.905472Z",
-     "start_time": "2022-02-18T13:39:40.767553Z"
-    },
-    "scrolled": false,
     "tags": [
      "nbval-ignore-output"
     ]
diff --git a/poetry.lock b/poetry.lock
index 0d11749e7..44add4054 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -220,7 +220,7 @@ unicode_backport = ["unicodedata2"]
 
 [[package]]
 name = "click"
-version = "8.0.3"
+version = "8.0.4"
 description = "Composable command line interface toolkit"
 category = "dev"
 optional = false
@@ -240,7 +240,7 @@ python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*"
 
 [[package]]
 name = "coverage"
-version = "6.3.1"
+version = "6.3.2"
 description = "Code coverage measurement for Python"
 category = "dev"
 optional = false
@@ -859,6 +859,17 @@ traitlets = ">=4.1"
 fast = ["fastjsonschema"]
 test = ["check-manifest", "fastjsonschema", "testpath", "pytest", "pytest-cov"]
 
+[[package]]
+name = "nbstripout"
+version = "0.5.0"
+description = "Strips outputs from Jupyter and IPython notebooks"
+category = "main"
+optional = false
+python-versions = ">=3.5"
+
+[package.dependencies]
+nbformat = "*"
+
 [[package]]
 name = "nbval"
 version = "0.9.6"
@@ -1052,7 +1063,7 @@ python-versions = ">=3.7"
 
 [[package]]
 name = "platformdirs"
-version = "2.5.0"
+version = "2.5.1"
 description = "A small Python module for determining appropriate platform-specific dirs, e.g. a \"user data dir\"."
 category = "dev"
 optional = false
@@ -1775,7 +1786,7 @@ scarlet = ["peigen", "autograd", "pybind11", "proxmin"]
 [metadata]
 lock-version = "1.1"
 python-versions = ">= 3.7.1, < 3.9.0"
-content-hash = "dd29d81eb65dfecac8ec7e09f7d3104b062c03f0e4f7694fa3472e4df70dbd8e"
+content-hash = "3b741e07eb889153af59cf0933b13ec51cd962e956732aff94773fc6d7d21146"
 
 [metadata.files]
 alabaster = [
@@ -1883,6 +1894,7 @@ black = [
     {file = "black-22.1.0.tar.gz", hash = "sha256:a7c0192d35635f6fc1174be575cb7915e92e5dd629ee79fdaf0dcfa41a80afb5"},
 ]
 bleach = [
+    {file = "bleach-4.1.0-py2.py3-none-any.whl", hash = "sha256:4d2651ab93271d1129ac9cbc679f524565cc8a1b791909c4a51eac4446a15994"},
     {file = "bleach-4.1.0.tar.gz", hash = "sha256:0900d8b37eba61a802ee40ac0061f8c2b5dee29c1927dd1d233e075ebf5a71da"},
 ]
 cached-property = [
@@ -1954,55 +1966,55 @@ charset-normalizer = [
     {file = "charset_normalizer-2.0.12-py3-none-any.whl", hash = "sha256:6881edbebdb17b39b4eaaa821b438bf6eddffb4468cf344f09f89def34a8b1df"},
 ]
 click = [
-    {file = "click-8.0.3-py3-none-any.whl", hash = "sha256:353f466495adaeb40b6b5f592f9f91cb22372351c84caeb068132442a4518ef3"},
-    {file = "click-8.0.3.tar.gz", hash = "sha256:410e932b050f5eed773c4cda94de75971c89cdb3155a72a0831139a79e5ecb5b"},
+    {file = "click-8.0.4-py3-none-any.whl", hash = "sha256:6a7a62563bbfabfda3a38f3023a1db4a35978c0abd76f6c9605ecd6554d6d9b1"},
+    {file = "click-8.0.4.tar.gz", hash = "sha256:8458d7b1287c5fb128c90e23381cf99dcde74beaf6c7ff6384ce84d6fe090adb"},
 ]
 colorama = [
     {file = "colorama-0.4.4-py2.py3-none-any.whl", hash = "sha256:9f47eda37229f68eee03b24b9748937c7dc3868f906e8ba69fbcbdd3bc5dc3e2"},
     {file = "colorama-0.4.4.tar.gz", hash = "sha256:5941b2b48a20143d2267e95b1c2a7603ce057ee39fd88e7329b0c292aa16869b"},
 ]
 coverage = [
-    {file = "coverage-6.3.1-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:eeffd96882d8c06d31b65dddcf51db7c612547babc1c4c5db6a011abe9798525"},
-    {file = "coverage-6.3.1-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:621f6ea7260ea2ffdaec64fe5cb521669984f567b66f62f81445221d4754df4c"},
-    {file = "coverage-6.3.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:84f2436d6742c01136dd940ee158bfc7cf5ced3da7e4c949662b8703b5cd8145"},
-    {file = "coverage-6.3.1-cp310-cp310-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:de73fca6fb403dd72d4da517cfc49fcf791f74eee697d3219f6be29adf5af6ce"},
-    {file = "coverage-6.3.1-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:78fbb2be068a13a5d99dce9e1e7d168db880870f7bc73f876152130575bd6167"},
-    {file = "coverage-6.3.1-cp310-cp310-musllinux_1_1_aarch64.whl", hash = "sha256:f5a4551dfd09c3bd12fca8144d47fe7745275adf3229b7223c2f9e29a975ebda"},
-    {file = "coverage-6.3.1-cp310-cp310-musllinux_1_1_i686.whl", hash = "sha256:7bff3a98f63b47464480de1b5bdd80c8fade0ba2832c9381253c9b74c4153c27"},
-    {file = "coverage-6.3.1-cp310-cp310-musllinux_1_1_x86_64.whl", hash = "sha256:a06c358f4aed05fa1099c39decc8022261bb07dfadc127c08cfbd1391b09689e"},
-    {file = "coverage-6.3.1-cp310-cp310-win32.whl", hash = "sha256:9fff3ff052922cb99f9e52f63f985d4f7a54f6b94287463bc66b7cdf3eb41217"},
-    {file = "coverage-6.3.1-cp310-cp310-win_amd64.whl", hash = "sha256:276b13cc085474e482566c477c25ed66a097b44c6e77132f3304ac0b039f83eb"},
-    {file = "coverage-6.3.1-cp37-cp37m-macosx_10_9_x86_64.whl", hash = "sha256:56c4a409381ddd7bbff134e9756077860d4e8a583d310a6f38a2315b9ce301d0"},
-    {file = "coverage-6.3.1-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:9eb494070aa060ceba6e4bbf44c1bc5fa97bfb883a0d9b0c9049415f9e944793"},
-    {file = "coverage-6.3.1-cp37-cp37m-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:5e15d424b8153756b7c903bde6d4610be0c3daca3986173c18dd5c1a1625e4cd"},
-    {file = "coverage-6.3.1-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:61d47a897c1e91f33f177c21de897267b38fbb45f2cd8e22a710bcef1df09ac1"},
-    {file = "coverage-6.3.1-cp37-cp37m-musllinux_1_1_aarch64.whl", hash = "sha256:25e73d4c81efa8ea3785274a2f7f3bfbbeccb6fcba2a0bdd3be9223371c37554"},
-    {file = "coverage-6.3.1-cp37-cp37m-musllinux_1_1_i686.whl", hash = "sha256:fac0bcc5b7e8169bffa87f0dcc24435446d329cbc2b5486d155c2e0f3b493ae1"},
-    {file = "coverage-6.3.1-cp37-cp37m-musllinux_1_1_x86_64.whl", hash = "sha256:72128176fea72012063200b7b395ed8a57849282b207321124d7ff14e26988e8"},
-    {file = "coverage-6.3.1-cp37-cp37m-win32.whl", hash = "sha256:1bc6d709939ff262fd1432f03f080c5042dc6508b6e0d3d20e61dd045456a1a0"},
-    {file = "coverage-6.3.1-cp37-cp37m-win_amd64.whl", hash = "sha256:618eeba986cea7f621d8607ee378ecc8c2504b98b3fdc4952b30fe3578304687"},
-    {file = "coverage-6.3.1-cp38-cp38-macosx_10_9_x86_64.whl", hash = "sha256:d5ed164af5c9078596cfc40b078c3b337911190d3faeac830c3f1274f26b8320"},
-    {file = "coverage-6.3.1-cp38-cp38-macosx_11_0_arm64.whl", hash = "sha256:352c68e233409c31048a3725c446a9e48bbff36e39db92774d4f2380d630d8f8"},
-    {file = "coverage-6.3.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:448d7bde7ceb6c69e08474c2ddbc5b4cd13c9e4aa4a717467f716b5fc938a734"},
-    {file = "coverage-6.3.1-cp38-cp38-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:9fde6b90889522c220dd56a670102ceef24955d994ff7af2cb786b4ba8fe11e4"},
-    {file = "coverage-6.3.1-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:e647a0be741edbb529a72644e999acb09f2ad60465f80757da183528941ff975"},
-    {file = "coverage-6.3.1-cp38-cp38-musllinux_1_1_aarch64.whl", hash = "sha256:6a5cdc3adb4f8bb8d8f5e64c2e9e282bc12980ef055ec6da59db562ee9bdfefa"},
-    {file = "coverage-6.3.1-cp38-cp38-musllinux_1_1_i686.whl", hash = "sha256:2dd70a167843b4b4b2630c0c56f1b586fe965b4f8ac5da05b6690344fd065c6b"},
-    {file = "coverage-6.3.1-cp38-cp38-musllinux_1_1_x86_64.whl", hash = "sha256:9ad0a117b8dc2061ce9461ea4c1b4799e55edceb236522c5b8f958ce9ed8fa9a"},
-    {file = "coverage-6.3.1-cp38-cp38-win32.whl", hash = "sha256:e92c7a5f7d62edff50f60a045dc9542bf939758c95b2fcd686175dd10ce0ed10"},
-    {file = "coverage-6.3.1-cp38-cp38-win_amd64.whl", hash = "sha256:482fb42eea6164894ff82abbcf33d526362de5d1a7ed25af7ecbdddd28fc124f"},
-    {file = "coverage-6.3.1-cp39-cp39-macosx_10_9_x86_64.whl", hash = "sha256:c5b81fb37db76ebea79aa963b76d96ff854e7662921ce742293463635a87a78d"},
-    {file = "coverage-6.3.1-cp39-cp39-macosx_11_0_arm64.whl", hash = "sha256:a4f923b9ab265136e57cc14794a15b9dcea07a9c578609cd5dbbfff28a0d15e6"},
-    {file = "coverage-6.3.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:56d296cbc8254a7dffdd7bcc2eb70be5a233aae7c01856d2d936f5ac4e8ac1f1"},
-    {file = "coverage-6.3.1-cp39-cp39-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:1245ab82e8554fa88c4b2ab1e098ae051faac5af829efdcf2ce6b34dccd5567c"},
-    {file = "coverage-6.3.1-cp39-cp39-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:3f2b05757c92ad96b33dbf8e8ec8d4ccb9af6ae3c9e9bd141c7cc44d20c6bcba"},
-    {file = "coverage-6.3.1-cp39-cp39-musllinux_1_1_aarch64.whl", hash = "sha256:9e3dd806f34de38d4c01416344e98eab2437ac450b3ae39c62a0ede2f8b5e4ed"},
-    {file = "coverage-6.3.1-cp39-cp39-musllinux_1_1_i686.whl", hash = "sha256:d651fde74a4d3122e5562705824507e2f5b2d3d57557f1916c4b27635f8fbe3f"},
-    {file = "coverage-6.3.1-cp39-cp39-musllinux_1_1_x86_64.whl", hash = "sha256:704f89b87c4f4737da2860695a18c852b78ec7279b24eedacab10b29067d3a38"},
-    {file = "coverage-6.3.1-cp39-cp39-win32.whl", hash = "sha256:2aed4761809640f02e44e16b8b32c1a5dee5e80ea30a0ff0912158bde9c501f2"},
-    {file = "coverage-6.3.1-cp39-cp39-win_amd64.whl", hash = "sha256:9976fb0a5709988778ac9bc44f3d50fccd989987876dfd7716dee28beed0a9fa"},
-    {file = "coverage-6.3.1-pp36.pp37.pp38-none-any.whl", hash = "sha256:463e52616ea687fd323888e86bf25e864a3cc6335a043fad6bbb037dbf49bbe2"},
-    {file = "coverage-6.3.1.tar.gz", hash = "sha256:6c3f6158b02ac403868eea390930ae64e9a9a2a5bbfafefbb920d29258d9f2f8"},
+    {file = "coverage-6.3.2-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:9b27d894748475fa858f9597c0ee1d4829f44683f3813633aaf94b19cb5453cf"},
+    {file = "coverage-6.3.2-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:37d1141ad6b2466a7b53a22e08fe76994c2d35a5b6b469590424a9953155afac"},
+    {file = "coverage-6.3.2-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:f9987b0354b06d4df0f4d3e0ec1ae76d7ce7cbca9a2f98c25041eb79eec766f1"},
+    {file = "coverage-6.3.2-cp310-cp310-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:26e2deacd414fc2f97dd9f7676ee3eaecd299ca751412d89f40bc01557a6b1b4"},
+    {file = "coverage-6.3.2-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:4dd8bafa458b5c7d061540f1ee9f18025a68e2d8471b3e858a9dad47c8d41903"},
+    {file = "coverage-6.3.2-cp310-cp310-musllinux_1_1_aarch64.whl", hash = "sha256:46191097ebc381fbf89bdce207a6c107ac4ec0890d8d20f3360345ff5976155c"},
+    {file = "coverage-6.3.2-cp310-cp310-musllinux_1_1_i686.whl", hash = "sha256:6f89d05e028d274ce4fa1a86887b071ae1755082ef94a6740238cd7a8178804f"},
+    {file = "coverage-6.3.2-cp310-cp310-musllinux_1_1_x86_64.whl", hash = "sha256:58303469e9a272b4abdb9e302a780072c0633cdcc0165db7eec0f9e32f901e05"},
+    {file = "coverage-6.3.2-cp310-cp310-win32.whl", hash = "sha256:2fea046bfb455510e05be95e879f0e768d45c10c11509e20e06d8fcaa31d9e39"},
+    {file = "coverage-6.3.2-cp310-cp310-win_amd64.whl", hash = "sha256:a2a8b8bcc399edb4347a5ca8b9b87e7524c0967b335fbb08a83c8421489ddee1"},
+    {file = "coverage-6.3.2-cp37-cp37m-macosx_10_9_x86_64.whl", hash = "sha256:f1555ea6d6da108e1999b2463ea1003fe03f29213e459145e70edbaf3e004aaa"},
+    {file = "coverage-6.3.2-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:e5f4e1edcf57ce94e5475fe09e5afa3e3145081318e5fd1a43a6b4539a97e518"},
+    {file = "coverage-6.3.2-cp37-cp37m-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:7a15dc0a14008f1da3d1ebd44bdda3e357dbabdf5a0b5034d38fcde0b5c234b7"},
+    {file = "coverage-6.3.2-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:21b7745788866028adeb1e0eca3bf1101109e2dc58456cb49d2d9b99a8c516e6"},
+    {file = "coverage-6.3.2-cp37-cp37m-musllinux_1_1_aarch64.whl", hash = "sha256:8ce257cac556cb03be4a248d92ed36904a59a4a5ff55a994e92214cde15c5bad"},
+    {file = "coverage-6.3.2-cp37-cp37m-musllinux_1_1_i686.whl", hash = "sha256:b0be84e5a6209858a1d3e8d1806c46214e867ce1b0fd32e4ea03f4bd8b2e3359"},
+    {file = "coverage-6.3.2-cp37-cp37m-musllinux_1_1_x86_64.whl", hash = "sha256:acf53bc2cf7282ab9b8ba346746afe703474004d9e566ad164c91a7a59f188a4"},
+    {file = "coverage-6.3.2-cp37-cp37m-win32.whl", hash = "sha256:8bdde1177f2311ee552f47ae6e5aa7750c0e3291ca6b75f71f7ffe1f1dab3dca"},
+    {file = "coverage-6.3.2-cp37-cp37m-win_amd64.whl", hash = "sha256:b31651d018b23ec463e95cf10070d0b2c548aa950a03d0b559eaa11c7e5a6fa3"},
+    {file = "coverage-6.3.2-cp38-cp38-macosx_10_9_x86_64.whl", hash = "sha256:07e6db90cd9686c767dcc593dff16c8c09f9814f5e9c51034066cad3373b914d"},
+    {file = "coverage-6.3.2-cp38-cp38-macosx_11_0_arm64.whl", hash = "sha256:2c6dbb42f3ad25760010c45191e9757e7dce981cbfb90e42feef301d71540059"},
+    {file = "coverage-6.3.2-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:c76aeef1b95aff3905fb2ae2d96e319caca5b76fa41d3470b19d4e4a3a313512"},
+    {file = "coverage-6.3.2-cp38-cp38-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:8cf5cfcb1521dc3255d845d9dca3ff204b3229401994ef8d1984b32746bb45ca"},
+    {file = "coverage-6.3.2-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:8fbbdc8d55990eac1b0919ca69eb5a988a802b854488c34b8f37f3e2025fa90d"},
+    {file = "coverage-6.3.2-cp38-cp38-musllinux_1_1_aarch64.whl", hash = "sha256:ec6bc7fe73a938933d4178c9b23c4e0568e43e220aef9472c4f6044bfc6dd0f0"},
+    {file = "coverage-6.3.2-cp38-cp38-musllinux_1_1_i686.whl", hash = "sha256:9baff2a45ae1f17c8078452e9e5962e518eab705e50a0aa8083733ea7d45f3a6"},
+    {file = "coverage-6.3.2-cp38-cp38-musllinux_1_1_x86_64.whl", hash = "sha256:fd9e830e9d8d89b20ab1e5af09b32d33e1a08ef4c4e14411e559556fd788e6b2"},
+    {file = "coverage-6.3.2-cp38-cp38-win32.whl", hash = "sha256:f7331dbf301b7289013175087636bbaf5b2405e57259dd2c42fdcc9fcc47325e"},
+    {file = "coverage-6.3.2-cp38-cp38-win_amd64.whl", hash = "sha256:68353fe7cdf91f109fc7d474461b46e7f1f14e533e911a2a2cbb8b0fc8613cf1"},
+    {file = "coverage-6.3.2-cp39-cp39-macosx_10_9_x86_64.whl", hash = "sha256:b78e5afb39941572209f71866aa0b206c12f0109835aa0d601e41552f9b3e620"},
+    {file = "coverage-6.3.2-cp39-cp39-macosx_11_0_arm64.whl", hash = "sha256:4e21876082ed887baed0146fe222f861b5815455ada3b33b890f4105d806128d"},
+    {file = "coverage-6.3.2-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:34626a7eee2a3da12af0507780bb51eb52dca0e1751fd1471d0810539cefb536"},
+    {file = "coverage-6.3.2-cp39-cp39-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:1ebf730d2381158ecf3dfd4453fbca0613e16eaa547b4170e2450c9707665ce7"},
+    {file = "coverage-6.3.2-cp39-cp39-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:dd6fe30bd519694b356cbfcaca9bd5c1737cddd20778c6a581ae20dc8c04def2"},
+    {file = "coverage-6.3.2-cp39-cp39-musllinux_1_1_aarch64.whl", hash = "sha256:96f8a1cb43ca1422f36492bebe63312d396491a9165ed3b9231e778d43a7fca4"},
+    {file = "coverage-6.3.2-cp39-cp39-musllinux_1_1_i686.whl", hash = "sha256:dd035edafefee4d573140a76fdc785dc38829fe5a455c4bb12bac8c20cfc3d69"},
+    {file = "coverage-6.3.2-cp39-cp39-musllinux_1_1_x86_64.whl", hash = "sha256:5ca5aeb4344b30d0bec47481536b8ba1181d50dbe783b0e4ad03c95dc1296684"},
+    {file = "coverage-6.3.2-cp39-cp39-win32.whl", hash = "sha256:f5fa5803f47e095d7ad8443d28b01d48c0359484fec1b9d8606d0e3282084bc4"},
+    {file = "coverage-6.3.2-cp39-cp39-win_amd64.whl", hash = "sha256:9548f10d8be799551eb3a9c74bbf2b4934ddb330e08a73320123c07f95cc2d92"},
+    {file = "coverage-6.3.2-pp36.pp37.pp38-none-any.whl", hash = "sha256:18d520c6860515a771708937d2f78f63cc47ab3b80cb78e86573b0a760161faf"},
+    {file = "coverage-6.3.2.tar.gz", hash = "sha256:03e2a7826086b91ef345ff18742ee9fc47a6839ccd517061ef8fa1976e652ce9"},
 ]
 cycler = [
     {file = "cycler-0.11.0-py3-none-any.whl", hash = "sha256:3a27e95f763a428a739d2add979fa7494c912a32c17c4c38c4d5f082cad165a3"},
@@ -2376,6 +2388,10 @@ nbformat = [
     {file = "nbformat-5.1.3-py3-none-any.whl", hash = "sha256:eb8447edd7127d043361bc17f2f5a807626bc8e878c7709a1c647abda28a9171"},
     {file = "nbformat-5.1.3.tar.gz", hash = "sha256:b516788ad70771c6250977c1374fcca6edebe6126fd2adb5a69aa5c2356fd1c8"},
 ]
+nbstripout = [
+    {file = "nbstripout-0.5.0-py2.py3-none-any.whl", hash = "sha256:1e8a471780d1429aaa5d954c4d26c312df8666cc84de0ff160d62d8b576ae65b"},
+    {file = "nbstripout-0.5.0.tar.gz", hash = "sha256:86ab50136998d62c9fa92478d2eb9ddc4137e51a28568f78fa8f24a6fbb6a7d8"},
+]
 nbval = [
     {file = "nbval-0.9.6-py2.py3-none-any.whl", hash = "sha256:4f9b780997d8942408853513f2c5ee6c1863de193559fc3f95e1c1cde8110439"},
     {file = "nbval-0.9.6.tar.gz", hash = "sha256:cfefcd2ef66ee2d337d0b252c6bcec4023384eb32e8b9e5fcc3ac80ab8cd7d40"},
@@ -2487,6 +2503,9 @@ pickleshare = [
     {file = "pickleshare-0.7.5.tar.gz", hash = "sha256:87683d47965c1da65cdacaf31c8441d12b8044cdec9aca500cd78fc2c683afca"},
 ]
 pillow = [
+    {file = "Pillow-9.0.1-1-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:a5d24e1d674dd9d72c66ad3ea9131322819ff86250b30dc5821cbafcfa0b96b4"},
+    {file = "Pillow-9.0.1-1-cp38-cp38-macosx_11_0_arm64.whl", hash = "sha256:2632d0f846b7c7600edf53c48f8f9f1e13e62f66a6dbc15191029d950bfed976"},
+    {file = "Pillow-9.0.1-1-cp39-cp39-macosx_11_0_arm64.whl", hash = "sha256:b9618823bd237c0d2575283f2939655f54d51b4527ec3972907a927acbcc5bfc"},
     {file = "Pillow-9.0.1-cp310-cp310-macosx_10_10_universal2.whl", hash = "sha256:9bfdb82cdfeccec50aad441afc332faf8606dfa5e8efd18a6692b5d6e79f00fd"},
     {file = "Pillow-9.0.1-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:5100b45a4638e3c00e4d2320d3193bdabb2d75e79793af7c3eb139e4f569f16f"},
     {file = "Pillow-9.0.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:528a2a692c65dd5cafc130de286030af251d2ee0483a5bf50c9348aefe834e8a"},
@@ -2521,8 +2540,8 @@ pillow = [
     {file = "Pillow-9.0.1.tar.gz", hash = "sha256:6c8bc8238a7dfdaf7a75f5ec5a663f4173f8c367e5a39f87e720495e1eed75fa"},
 ]
 platformdirs = [
-    {file = "platformdirs-2.5.0-py3-none-any.whl", hash = "sha256:30671902352e97b1eafd74ade8e4a694782bd3471685e78c32d0fdfd3aa7e7bb"},
-    {file = "platformdirs-2.5.0.tar.gz", hash = "sha256:8ec11dfba28ecc0715eb5fb0147a87b1bf325f349f3da9aab2cd6b50b96b692b"},
+    {file = "platformdirs-2.5.1-py3-none-any.whl", hash = "sha256:bcae7cab893c2d310a711b70b24efb93334febe65f8de776ee320b517471e227"},
+    {file = "platformdirs-2.5.1.tar.gz", hash = "sha256:7535e70dfa32e84d4b34996ea99c5e432fa29a708d0f4e394bbcb2a8faa4f16d"},
 ]
 pluggy = [
     {file = "pluggy-1.0.0-py2.py3-none-any.whl", hash = "sha256:74134bbf457f031a36d68416e1509f34bd5ccc019f0bcc952c7b909d06b37bd3"},
diff --git a/pyproject.toml b/pyproject.toml
index f071e2ef9..3b4945e49 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -33,6 +33,7 @@ galsim = ">=2.2.4"
 hydra-core = ">=1.0.6"
 ipywidgets = ">=7.6.5"
 matplotlib = ">=3.4.3"
+nbstripout = "^0.5.0"
 numpy = ">=1.20.1"
 pandas = ">=1.3.3"
 peigen = {version = ">=0.0.9", optional = true}
@@ -47,7 +48,7 @@ tqdm = "^4.62.3"
 
 [tool.poetry.dev-dependencies]
 Cython = ">=0.29.24"
-black = ">=20.8b1"
+black = ">=22.1.0"
 flake8 = ">=3.9.2"
 flake8-absolute-import = ">=1.0"
 jupyter-sphinx = ">=0.3.2"