diff --git a/qim3d/__init__.py b/qim3d/__init__.py
index beadf9df2d278b6ed9a99d1bb2d4989e22d7dd41..c89b6fd07af948ddd1d126632e2713dd1c6e6f0e 100644
--- a/qim3d/__init__.py
+++ b/qim3d/__init__.py
@@ -1,3 +1,15 @@
+"""qim3d: A Python package for 3D image processing and visualization.
+
+The qim3d library is designed to make it easier to work with 3D imaging data in Python. 
+It offers a range of features, including data loading and manipulation,
+ image processing and filtering, visualization of 3D data, and analysis of imaging results.
+
+Documentation available at https://platform.qim.dk/qim3d/
+
+"""
+
+__version__ = "0.3.6"
+
 import logging
 
 logging.basicConfig(level=logging.ERROR)
@@ -5,12 +17,11 @@ logging.basicConfig(level=logging.ERROR)
 from . import io
 from . import gui
 from . import viz
-
 from . import utils
-from . import models
 from . import processing
 
+# Commenting out models because it takes too long to import
+# from . import models
 
-__version__ = "0.3.5"
 examples = io.ImgExamples()
 io.logger.set_level_info()
diff --git a/qim3d/io/loading.py b/qim3d/io/loading.py
index 1b986b7a3a9a943be8a28b4fa8cbcc9d4f464dc8..42eef66c7e73dcda4619c83690a74c09c30eb528 100644
--- a/qim3d/io/loading.py
+++ b/qim3d/io/loading.py
@@ -31,7 +31,7 @@ from qim3d.io.logger import log
 from qim3d.utils.internal_tools import get_file_size, sizeof, stringify_path
 from qim3d.utils.system import Memory
 
-dask.config.set(scheduler="processes")  # Dask parallel goes brrrrr
+dask.config.set(scheduler="processes") 
 
 
 class DataLoader:
diff --git a/qim3d/utils/cli.py b/qim3d/utils/cli.py
index 2280f3af66d810172c91b57f948d123a22d92010..92f01e3388a935daad8021bc58664a1aa9512ce9 100644
--- a/qim3d/utils/cli.py
+++ b/qim3d/utils/cli.py
@@ -1,10 +1,11 @@
 import argparse
 import webbrowser
 
-import qim3d
 from qim3d.gui import annotation_tool, data_explorer, iso3d, local_thickness
 from qim3d.io.loading import DataLoader
 from qim3d.utils import image_preview
+from qim3d import __version__ as version
+import qim3d
 
 
 def main():
@@ -150,7 +151,7 @@ def main():
             "--- Welcome to qim3d command-line interface ---\n"
             "qim3d is a Python package for 3D image processing and visualization.\n"
             "For more information, please visit: https://platform.qim.dk/qim3d/\n"
-            f"Current version of qim3d: {qim3d.__version__}\n"
+            f"Current version of qim3d: {version}\n"
             " \n"
             "The qim3d command-line interface provides the following subcommands:\n"
             "- gui: Graphical User Interfaces\n"
diff --git a/qim3d/viz/__init__.py b/qim3d/viz/__init__.py
index 96352d2bef9d38dfd816d3500d240fb3ff838890..080a4ebd7a812cb659c5b817d18767a3d287568a 100644
--- a/qim3d/viz/__init__.py
+++ b/qim3d/viz/__init__.py
@@ -1,4 +1,3 @@
-#from .colormaps import objects
 from . import colormaps
 from .cc import plot_cc
 from .detection import circles
diff --git a/qim3d/viz/colormaps.py b/qim3d/viz/colormaps.py
index d72c94d9ec1f104ec605f487b89dbbdf9b5cc609..547e10fb8f1afd0e70b1993d9c73fc7c902f4683 100644
--- a/qim3d/viz/colormaps.py
+++ b/qim3d/viz/colormaps.py
@@ -10,8 +10,6 @@ from matplotlib.colors import LinearSegmentedColormap
 from matplotlib import colormaps
 from skimage import color
 
-from qim3d.io.logger import log
-
 def rearrange_colors(randRGBcolors_old, min_dist = 0.5):
     # Create new list for re-arranged colors
     randRGBcolors_new = [randRGBcolors_old.pop(0)]
diff --git a/setup.py b/setup.py
index 40caf8e9001803be246912181eb9ba47a18995d1..017d277a0acb04bd4f63301c31186d0432e97d9b 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@ with open("README.md", "r", encoding="utf-8") as f:
 
 setup(
     name="qim3d",
-    version="0.3.5",
+    version="0.3.6",
     author="Felipe Delestro",
     author_email="fima@dtu.dk",
     description="QIM tools and user interfaces",