diff --git a/diamond/__init__.py b/diamond/__init__.py
index 36c6e9b..c35d76a 100644
--- a/diamond/__init__.py
+++ b/diamond/__init__.py
@@ -1 +1,105 @@
-from diamond.diamond import *
+from diamond import diamond
+from diamond.diamond import (
+    K3,
+    HochschildHomologies,
+    HochschildHomology,
+    HodgeDiamond,
+    HodgeDiamondRing,
+    K3n,
+    Mzeronbar,
+    Pn,
+    abelian,
+    complete_intersection,
+    curve,
+    cyclic_cover,
+    enriques,
+    fano_threefold,
+    fano_variety_intersection_quadrics_even,
+    fano_variety_intersection_quadrics_odd,
+    fano_variety_lines_cubic,
+    generalised_grassmannian,
+    generalised_kummer,
+    grassmannian,
+    gushel_mukai,
+    hilbn,
+    hilbthree,
+    hilbtwo,
+    horospherical,
+    hypersurface,
+    jacobian,
+    kummer_resolution,
+    lagrangian_grassmannian,
+    lefschetz,
+    moduli_parabolic_vector_bundles_rank_two,
+    moduli_vector_bundles,
+    nestedhilbn,
+    odd_symplectic_grassmannian,
+    ogrady6,
+    ogrady10,
+    orthogonal_grassmannian,
+    partial_flag_variety,
+    point,
+    quiver_moduli,
+    quot_scheme_curve,
+    ruled,
+    seshadris_desingularisation,
+    slope,
+    surface,
+    symmetric_power,
+    symplectic_grassmannian,
+    weighted_hypersurface,
+    zero,
+)
+
+__all__ = [
+    "HochschildHomologies",
+    "HochschildHomology",
+    "HodgeDiamond",
+    "HodgeDiamondRing",
+    "K3",
+    "K3n",
+    "Mzeronbar",
+    "Pn",
+    "abelian",
+    "complete_intersection",
+    "curve",
+    "cyclic_cover",
+    "diamond",
+    "enriques",
+    "fano_threefold",
+    "fano_variety_intersection_quadrics_even",
+    "fano_variety_intersection_quadrics_odd",
+    "fano_variety_lines_cubic",
+    "generalised_grassmannian",
+    "generalised_kummer",
+    "grassmannian",
+    "gushel_mukai",
+    "hilbn",
+    "hilbthree",
+    "hilbtwo",
+    "horospherical",
+    "hypersurface",
+    "jacobian",
+    "kummer_resolution",
+    "lagrangian_grassmannian",
+    "lefschetz",
+    "moduli_parabolic_vector_bundles_rank_two",
+    "moduli_vector_bundles",
+    "nestedhilbn",
+    "odd_symplectic_grassmannian",
+    "ogrady10",
+    "ogrady6",
+    "orthogonal_grassmannian",
+    "partial_flag_variety",
+    "point",
+    "quiver_moduli",
+    "quot_scheme_curve",
+    "ruled",
+    "seshadris_desingularisation",
+    "slope",
+    "surface",
+    "symmetric_power",
+    "symplectic_grassmannian",
+    "weighted_hypersurface",
+    "zero",
+]