diff --git a/src/toolbox_scs/base/__init__.py b/src/toolbox_scs/base/__init__.py
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..367a07d36ff6de9fde66910bfe28d8b7a2f83652 100644
--- a/src/toolbox_scs/base/__init__.py
+++ b/src/toolbox_scs/base/__init__.py
@@ -0,0 +1,5 @@
+from . import knife_edge as knife_edge_module
+from .knife_edge import *
+
+
+__all__ = knife_edge_module.__all__
diff --git a/src/toolbox_scs/base/knife_edge.py b/src/toolbox_scs/base/knife_edge.py
index adabfd49660ff1a75e920a92d47f87e373c838d3..7e9fbca8e5dc30619746879af11d6ada56cd5052 100644
--- a/src/toolbox_scs/base/knife_edge.py
+++ b/src/toolbox_scs/base/knife_edge.py
@@ -3,10 +3,11 @@ from scipy import special
 from scipy.optimize import curve_fit
 
 
+__all__ = ['knife_edge', 'knife_edge_base']
+
+
 def knife_edge(positions, intensities, axisRange=None, p0=None):
     """
-    The base implementation of the knife-edge scan analysis.
-
     Calculates the beam radius at 1/e^2 from a knife-edge scan by
     fitting with erfc function: f(a,b,u) = a*erfc(u) + b or
     where u = sqrt(2)*(x-x0)/w0 with w0 the beam radius at 1/e^2