diff --git a/docs/api.md b/docs/api.md
index 894e449..53921c7 100644
--- a/docs/api.md
+++ b/docs/api.md
@@ -20,8 +20,8 @@ The central class is `FlowSOM`, which can be instantiated as a class and is also
 .. autosummary::
     :toctree: generated
 
-    flowsom.FlowSOM
-    flowsom.flowsom_clustering
+    FlowSOM
+    flowsom_clustering
 ```
 
 ## Reading
diff --git a/src/FlowSOM/pl/__init__.py b/src/FlowSOM/pl/__init__.py
index 8f19807..23c98ae 100644
--- a/src/FlowSOM/pl/__init__.py
+++ b/src/FlowSOM/pl/__init__.py
@@ -10,4 +10,13 @@
     plot_star_legend,
     scale_star_heights,
 )
-from .plot_functions import FlowSOMmary, plot_2D_scatters, plot_marker, plot_pies, plot_stars, plot_variable
+from .plot_functions import (
+    FlowSOMmary,
+    plot_2D_scatters,
+    plot_labels,
+    plot_marker,
+    plot_numbers,
+    plot_pies,
+    plot_stars,
+    plot_variable,
+)