diff --git a/README.md b/README.md
index 834c65b..cfbf287 100644
--- a/README.md
+++ b/README.md
@@ -9,6 +9,13 @@ version: 0.0.1
 pip install sf-segmenter
 ```
 
+To use `sf_segmenter.vis`, matplotlib needs to be installed; you can install it with the `vis` extra:
+
+```
+pip install sf-segmenter[vis]
+```
+
+
 ## Reference
 * Serrà, J., Müller, M., Grosche, P., & Arcos, J. L. (2012). Unsupervised Detection of Music Boundaries by Time Series Structure Features. In Proc. of the 26th AAAI Conference on Artificial Intelligence (pp. 1613–1619).Toronto, Canada.
 
diff --git a/setup.py b/setup.py
index 499b000..f8a3d74 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,10 @@
     license='MIT',
     install_requires=[
         'miditoolkit >= 0.1.14',
-    ]
+    ],
+    extras_require={
+        'vis': ['matplotlib'],
+    },
 )
 
 
diff --git a/sf_segmenter/__init__.py b/sf_segmenter/__init__.py
index 0561766..d18f409 100644
--- a/sf_segmenter/__init__.py
+++ b/sf_segmenter/__init__.py
@@ -1,5 +1 @@
-from .feature import *
-from .segmenter import *
-from .vis import *
-
-__version__ = '0.0.2'
\ No newline at end of file
+__version__ = '0.0.2'