diff --git a/src/napari_segment/__init__.py b/src/napari_segment/__init__.py
index 6f90367354680864560ddd18abb8aed582b1eba6..7146cbaef5e9f35c0712ceea617e424cee1238fe 100644
--- a/src/napari_segment/__init__.py
+++ b/src/napari_segment/__init__.py
@@ -1,4 +1,3 @@
-
 try:
     from ._version import version as __version__
 except ImportError:
@@ -6,6 +5,15 @@ except ImportError:
 
 
 from ._reader import napari_get_reader
-from ._writer import write_single_image, write_multiple 
-from ._sample_data import make_sample_data 
+from ._sample_data import make_sample_data
 from ._widget import ExampleQWidget, example_magic_widget
+from ._writer import write_multiple, write_single_image
+
+__ALL__ = [
+    napari_get_reader,
+    write_multiple,
+    write_single_image,
+    make_sample_data,
+    ExampleQWidget,
+    example_magic_widget,
+]