diff --git a/pyproject.toml b/pyproject.toml
index adaba4c..630d26b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "salamander-learn"
-version = "0.3.2"
+version = "0.4.0"
 description = "Salamander is a non-negative matrix factorization framework for signature analysis"
 license = "MIT"
 authors = ["Benedikt Geiger"]
diff --git a/src/salamander/__init__.py b/src/salamander/__init__.py
index bad5934..35d6efa 100644
--- a/src/salamander/__init__.py
+++ b/src/salamander/__init__.py
@@ -7,11 +7,12 @@
 from . import plot as pl
 from . import tools as tl
 
-__version__ = "0.3.2"
+__version__ = "0.4.0"
 
 pl.set_salamander_style()
 
 __all__ = [
+    "__version__",
     "models",
     "pl",
     "tl",