diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 309e70b..1ac626e 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -23,8 +23,8 @@ jobs:
     # only run once if internal PR
     if: github.event_name == 'push' || github.event.pull_request.head.repo.full_name != github.repository
     strategy:
-        matrix:
-          include:
+      matrix:
+        include:
           - label: earliest
             os: ubuntu-latest
             python-version: 3.8
@@ -32,97 +32,117 @@ jobs:
             coverage: false
           - label: baseline
             os: ubuntu-latest
-            python-version: 3.9
+            python-version: "3.10"
             rdkit-version: "rdkit~=2022.09"
             coverage: true
           - label: latest
             os: ubuntu-latest
-            python-version: "3.10"
+            python-version: "3.12"
             rdkit-version: "rdkit"
             coverage: false
 
     steps:
-    - uses: actions/checkout@v4
-
-    - name: Setup Conda
-      uses: conda-incubator/setup-miniconda@v2
-      with:
-        python-version: ${{ matrix.python-version }}
-        auto-update-conda: true
-        channel-priority: flexible
-        channels: conda-forge, defaults
-        add-pip-as-python-dependency: true
-        architecture: x64
-        use-mamba: true
-        miniforge-variant: Mambaforge
-
-    - name: Check conda and pip
-      run: |
-        which python
-        python --version
-        pip --version
-        conda --version
-        mamba --version
-
-    - name: Install conda dependencies
-      run: |
-        mamba install ${{ matrix.rdkit-version }}
-        mamba list
-
-    - name: Install package through pip
-      run: |
-        pip install .[dev]
-        pip list
-
-    - name: Run tests
-      run: |
-        pytest --color=yes --disable-pytest-warnings --cov=prolif --cov-report=xml tests/
-
-    - name: Measure tests coverage
-      if: matrix.coverage
-      uses: codecov/codecov-action@v3
-      with:
-        files: ./coverage.xml
-        fail_ci_if_error: true
-        verbose: true
-        token: ${{ secrets.CODECOV_TOKEN }}
-
-    - name: Build
-      run: |
-        python -m build
-
-    - name: Remove previous ProLIF install
-      run: |
-        pip uninstall -y prolif
-
-    - name: Install from tar.gz
-      run: |
-        pip install dist/prolif-*.tar.gz
-
-    - name: Test tar.gz install
-      run: |
-        python <<EOF
-        import prolif
-        import os
-        print(prolif.__version__)
-        from prolif.plotting.network import LigNetwork
-        assert os.path.isfile(prolif.datafiles.TOP)
-        EOF
-
-    - name: Remove previous ProLIF install
-      run: |
-        pip uninstall -y prolif
-
-    - name: Install from wheel
-      run: |
-        pip install dist/prolif-*.whl
-
-    - name: Test wheel install
-      run: |
-        python <<EOF
-        import prolif
-        import os
-        print(prolif.__version__)
-        from prolif.plotting.network import LigNetwork
-        assert os.path.isfile(prolif.datafiles.TOP)
-        EOF
+      - uses: actions/checkout@v4
+
+      - name: Setup Conda
+        uses: conda-incubator/setup-miniconda@v2
+        with:
+          python-version: ${{ matrix.python-version }}
+          auto-update-conda: true
+          channel-priority: flexible
+          channels: conda-forge, defaults
+          add-pip-as-python-dependency: true
+          architecture: x64
+          use-mamba: true
+          miniforge-variant: Mambaforge
+
+      - name: Check conda and pip
+        run: |
+          which python
+          python --version
+          pip --version
+          conda --version
+          mamba --version
+
+      - name: Install conda dependencies
+        run: |
+          mamba install ${{ matrix.rdkit-version }}
+          mamba list
+
+      - name: Install package through pip
+        run: |
+          pip install .[dev]
+          pip list
+
+      - name: Build
+        run: |
+          python -m build
+
+      - name: Ensure tests and data included in source dist
+        run: |
+          tar -ztvf dist/prolif-*.tar.gz | grep -E 'prolif-.+/tests/.+' || exit 1
+          tar -ztvf dist/prolif-*.tar.gz | grep -E 'prolif-.+/data/.+' || exit 1
+          tar -ztvf dist/prolif-*.tar.gz | grep -E 'prolif-.+/data/vina/.+' || exit 1
+
+      - name: Run tests
+        run: |
+          pytest --color=yes --disable-pytest-warnings --cov=prolif --cov-report=xml tests/
+
+      - name: Measure tests coverage
+        if: matrix.coverage
+        uses: codecov/codecov-action@v3
+        with:
+          files: ./coverage.xml
+          fail_ci_if_error: true
+          verbose: true
+          token: ${{ secrets.CODECOV_TOKEN }}
+
+      - name: Build
+        run: |
+          python -m build
+
+      - name: Remove previous ProLIF install
+        run: |
+          pip uninstall -y prolif
+
+      - name: Install from tar.gz
+        run: |
+          pip install dist/prolif-*.tar.gz
+
+      - name: Test tar.gz install
+        working-directory: /
+        run: |
+          python <<EOF
+          from pathlib import Path
+          from contextlib import suppress
+          import prolif
+          print(prolif.__version__)
+          from prolif.plotting.network import LigNetwork
+          assert Path(prolif.datafiles.TOP).is_file()
+          with suppress(ImportError, ModuleNotFoundError):
+              import tests
+              assert next(Path(tests.__file__).parent.glob("test_fingerprint.py"), None) is None
+          EOF
+
+      - name: Remove previous ProLIF install
+        run: |
+          pip uninstall -y prolif
+
+      - name: Install from wheel
+        run: |
+          pip install dist/prolif-*.whl
+
+      - name: Test wheel install
+        working-directory: /
+        run: |
+          python <<EOF
+          from pathlib import Path
+          from contextlib import suppress
+          import prolif
+          print(prolif.__version__)
+          from prolif.plotting.network import LigNetwork
+          assert Path(prolif.datafiles.TOP).is_file()
+          with suppress(ImportError, ModuleNotFoundError):
+              import tests
+              assert next(Path(tests.__file__).parent.glob("test_fingerprint.py"), None) is None
+          EOF
diff --git a/CHANGELOG.md b/CHANGELOG.md
index 285512c..a6390ce 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,26 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [Unreleased]
 
 
+## [2.0.2] - 2024-03-01
+
+Maintenance update:
+
+### Added
+
+- `PROLIF_N_JOBS` environment variable to control multiprocessing during the
+  documentation build.
+
+### Changed
+
+- Updated CI pipelines to test on Python 3.8, 3.10 and 3.12.
+
+### Fixed
+
+- Packaging maintenance to fix the `tests/` directory being included as a module rather
+  than as plain data files. This could interfere with `pytest` during test collection
+  in other packages.
+
+
 ## [2.0.1] - 2023-10-11
 
 ### Added
diff --git a/docs/conf.py b/docs/conf.py
index 2b7161a..bc7ad12 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -54,7 +54,7 @@
 ]
 nb_execution_allow_errors = False
 nb_execution_raise_on_error = True
-nb_execution_timeout = 180
+nb_execution_timeout = 300
 
 mathjax_path = "https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.0/MathJax.js?config=TeX-AMS-MML_HTMLorMML"
 autosectionlabel_prefix_document = True
diff --git a/prolif/_version.py b/prolif/_version.py
index 159d48b..7652af9 100644
--- a/prolif/_version.py
+++ b/prolif/_version.py
@@ -1 +1 @@
-__version__ = "2.0.1"
+__version__ = "2.0.2-rc1"
diff --git a/prolif/fingerprint.py b/prolif/fingerprint.py
index cb16f7d..6a043e9 100644
--- a/prolif/fingerprint.py
+++ b/prolif/fingerprint.py
@@ -24,6 +24,8 @@
     fp = prolif.Fingerprint.from_pickle("fingerprint.pkl")
 
 """
+
+import os
 import warnings
 from collections.abc import Sized
 from functools import wraps
@@ -462,6 +464,8 @@ def run(
             raise ValueError("converter_kwargs must be a list of 2 dicts")
 
         converter_kwargs = converter_kwargs or ({}, {})
+        if n_jobs is None:
+            n_jobs = int(os.environ.get("PROLIF_N_JOBS", 0)) or None
         if residues == "all":
             residues = list(Molecule.from_mda(prot, **converter_kwargs[1]).residues)
         if n_jobs != 1:
diff --git a/prolif/ifp.py b/prolif/ifp.py
index 7935350..4f586ee 100644
--- a/prolif/ifp.py
+++ b/prolif/ifp.py
@@ -2,6 +2,7 @@
 Storing interactions --- :mod:`prolif.ifp`
 ==========================================
 """
+
 from collections import UserDict
 
 from prolif.residue import ResidueId
diff --git a/prolif/interactions/base.py b/prolif/interactions/base.py
index 6bbe237..0e2df25 100644
--- a/prolif/interactions/base.py
+++ b/prolif/interactions/base.py
@@ -4,6 +4,7 @@
 
 This module contains the base classes used to build most of the interactions.
 """
+
 import warnings
 from itertools import product
 from math import degrees, radians
diff --git a/prolif/interactions/interactions.py b/prolif/interactions/interactions.py
index f3f733f..7fe576c 100644
--- a/prolif/interactions/interactions.py
+++ b/prolif/interactions/interactions.py
@@ -64,9 +64,7 @@ class Hydrophobic(Distance):
 
     def __init__(
         self,
-        hydrophobic=(
-            "[c,s,Br,I,S&H0&v2," "$([D3,D4;#6])&!$([#6]~[#7,#8,#9])&!$([#6X4H0]);+0]"
-        ),
+        hydrophobic="[c,s,Br,I,S&H0&v2,$([D3,D4;#6])&!$([#6]~[#7,#8,#9])&!$([#6X4H0]);+0]",
         distance=4.5,
     ):
         super().__init__(
@@ -99,11 +97,7 @@ class HBAcceptor(SingleAngle):
 
     def __init__(
         self,
-        acceptor=(
-            "[#7&!$([nX3])&!$([NX3]-*=[O,N,P,S])&!$([NX3]-[a])&!$([Nv4&+1]),"
-            "O&!$([OX2](C)C=O)&!$(O(~a)~a)&!$(O=N-*)&!$([O-]-N=O),o+0,"
-            "F&$(F-[#6])&!$(F-[#6][F,Cl,Br,I])]"
-        ),
+        acceptor="[#7&!$([nX3])&!$([NX3]-*=[O,N,P,S])&!$([NX3]-[a])&!$([Nv4&+1]),O&!$([OX2](C)C=O)&!$(O(~a)~a)&!$(O=N-*)&!$([O-]-N=O),o+0,F&$(F-[#6])&!$(F-[#6][F,Cl,Br,I])]",
         donor="[$([O,S;+0]),$([N;v3,v4&+1]),n+0]-[H]",
         distance=3.5,
         DHA_angle=(130, 180),
diff --git a/prolif/molecule.py b/prolif/molecule.py
index f44a0ed..3d28d69 100644
--- a/prolif/molecule.py
+++ b/prolif/molecule.py
@@ -2,6 +2,7 @@
 Reading proteins and ligands --- :mod:`prolif.molecule`
 =======================================================
 """
+
 import copy
 from collections import defaultdict
 from collections.abc import Sequence
@@ -116,7 +117,7 @@ def from_mda(cls, obj, selection=None, **kwargs):
         """
         ag = obj.select_atoms(selection) if selection else obj.atoms
         if ag.n_atoms == 0:
-            raise mda.SelectionError(f"AtomGroup is empty, please check your selection")
+            raise mda.SelectionError("AtomGroup is empty, please check your selection")
         mol = ag.convert_to.rdkit(**kwargs)
         return cls(mol)
 
diff --git a/prolif/parallel.py b/prolif/parallel.py
index 3705087..c35dd51 100644
--- a/prolif/parallel.py
+++ b/prolif/parallel.py
@@ -7,6 +7,7 @@
 in :meth:`~prolif.fingerprint.Fingerprint.run` and
 :meth:`~prolif.fingerprint.Fingerprint.run_from_iterable` respectively.
 """
+
 from ctypes import c_uint32
 from threading import Event, Thread
 from time import sleep
diff --git a/prolif/pickling.py b/prolif/pickling.py
index 4b455d6..b97bc55 100644
--- a/prolif/pickling.py
+++ b/prolif/pickling.py
@@ -9,6 +9,7 @@
 behavior upon importing the `prolif` package to ensure that the parallelization code
 works as on other platforms.
 """
+
 import sys
 from typing import Optional
 
diff --git a/prolif/plotting/barcode.py b/prolif/plotting/barcode.py
index 225765b..046c61e 100644
--- a/prolif/plotting/barcode.py
+++ b/prolif/plotting/barcode.py
@@ -8,6 +8,7 @@
    :members:
 
 """
+
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, ClassVar, Dict, List, Literal, Optional, Tuple
@@ -73,9 +74,9 @@ def _bit_to_color_value(s: pd.Series) -> pd.Series:
             """Replaces a bit value with it's corresponding color value"""
             interaction = s.name[-1]
             return s.apply(
-                lambda v: self.color_mapper[interaction]
-                if v
-                else self.color_mapper[None]
+                lambda v: (
+                    self.color_mapper[interaction] if v else self.color_mapper[None]
+                )
             )
 
         self.df = df.astype(np.uint8).T.apply(_bit_to_color_value, axis=1)
@@ -85,7 +86,8 @@ def from_fingerprint(cls, fp: Fingerprint) -> Barcode:
         """Creates a barcode object from a fingerprint."""
         if not hasattr(fp, "ifp"):
             raise RunRequiredError(
-                "Please run the fingerprint analysis before attempting to display results."
+                "Please run the fingerprint analysis before attempting to display"
+                " results."
             )
         return cls(fp.to_dataframe())
 
diff --git a/prolif/plotting/complex3d.py b/prolif/plotting/complex3d.py
index a8573cc..26022e4 100644
--- a/prolif/plotting/complex3d.py
+++ b/prolif/plotting/complex3d.py
@@ -8,6 +8,7 @@
    :members:
 
 """
+
 from __future__ import annotations
 
 from copy import deepcopy
@@ -101,7 +102,8 @@ class Complex3D:
     PROTEIN_RING_INTERACTIONS: ClassVar[Set[str]] = {*RING_SYSTEMS, "CationPi"}
     RESIDUE_HOVER_CALLBACK: ClassVar[
         str
-    ] = """function(atom,viewer) {
+    ] = """
+    function(atom,viewer) {
         if(!atom.label) {
             atom.label = viewer.addLabel('%s:'+atom.atom+atom.serial,
                 {position: atom, backgroundColor: 'mintcream', fontColor:'black'});
@@ -157,7 +159,8 @@ def from_fingerprint(
         """
         if not hasattr(fp, "ifp"):
             raise RunRequiredError(
-                "Please run the fingerprint analysis before attempting to display results."
+                "Please run the fingerprint analysis before attempting to display"
+                " results."
             )
         ifp = fp.ifp[frame]
         return cls(ifp, lig_mol, prot_mol)
diff --git a/prolif/plotting/network.py b/prolif/plotting/network.py
index 2230693..c9e2e84 100644
--- a/prolif/plotting/network.py
+++ b/prolif/plotting/network.py
@@ -12,6 +12,7 @@
    :members:
 
 """
+
 import json
 import re
 import warnings
@@ -317,7 +318,8 @@ def from_fingerprint(
         """
         if not hasattr(fp, "ifp"):
             raise RunRequiredError(
-                "Please run the fingerprint analysis before attempting to display results."
+                "Please run the fingerprint analysis before attempting to display"
+                " results."
             )
         if kind == "frame":
             df = cls._make_frame_df_from_fp(fp, frame=frame, display_all=display_all)
diff --git a/prolif/plotting/residues.py b/prolif/plotting/residues.py
index 7ae68b5..ec3545e 100644
--- a/prolif/plotting/residues.py
+++ b/prolif/plotting/residues.py
@@ -7,6 +7,7 @@
 .. autofunction:: display_residues
 
 """
+
 from typing import Any, Optional, Tuple
 
 from rdkit import Chem
diff --git a/prolif/rdkitmol.py b/prolif/rdkitmol.py
index eed076c..5967b92 100644
--- a/prolif/rdkitmol.py
+++ b/prolif/rdkitmol.py
@@ -2,6 +2,7 @@
 Reading RDKit molecules --- :mod:`prolif.rdkitmol`
 ==================================================
 """
+
 from rdkit import Chem
 from rdkit.Chem.rdMolTransforms import ComputeCentroid
 
diff --git a/prolif/residue.py b/prolif/residue.py
index d646446..d92334e 100644
--- a/prolif/residue.py
+++ b/prolif/residue.py
@@ -2,6 +2,7 @@
 Residue-related classes --- :mod:`prolif.residue`
 =================================================
 """
+
 import re
 from collections import UserDict
 from typing import List, Optional
@@ -182,8 +183,7 @@ def __getitem__(self, key):
         # bool is a subclass of int but shouldn't be used here
         if isinstance(key, bool):
             raise KeyError(
-                "Expected a ResidueId, int, or str, "
-                f"got {type(key).__name__!r} instead"
+                f"Expected a ResidueId, int, or str, got {type(key).__name__!r} instead"
             )
         if isinstance(key, int):
             return self._residues[key]
@@ -193,7 +193,7 @@ def __getitem__(self, key):
         elif isinstance(key, ResidueId):
             return self.data[key]
         raise KeyError(
-            "Expected a ResidueId, int, or str, " f"got {type(key).__name__!r} instead"
+            f"Expected a ResidueId, int, or str, got {type(key).__name__!r} instead"
         )
 
     def select(self, mask):
diff --git a/prolif/utils.py b/prolif/utils.py
index c838135..65fa1d9 100644
--- a/prolif/utils.py
+++ b/prolif/utils.py
@@ -2,6 +2,7 @@
 Helper functions --- :mod:`prolif.utils`
 ========================================
 """
+
 import warnings
 from collections import defaultdict
 from contextlib import contextmanager
diff --git a/pyproject.toml b/pyproject.toml
index be1d8da..fa617e8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,6 +16,8 @@ classifiers = [
     "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
     "Topic :: Scientific/Engineering :: Bio-Informatics",
     "Topic :: Scientific/Engineering :: Chemistry",
 ]
@@ -76,14 +78,15 @@ Changelog = "https://github.com/chemosim-lab/ProLIF/blob/master/CHANGELOG.md"
 
 [tool.setuptools]
 zip-safe = false
-include-package-data = false
 
 [tool.setuptools.packages.find]
-include = ["prolif*", "tests*"]
+include = ["prolif*"]
 
 [tool.setuptools.package-data]
 "prolif.data" = ["*"]
 "prolif.data.vina" = ["*"]
+"tests" = ["*"]
+"tests.plotting" = ["*"]
 
 [tool.setuptools.dynamic]
 version = { attr = "prolif._version.__version__" }