diff --git a/docs/models.md b/docs/models.md
index 5e058b8eac5f8e435d8357153b5979824537ae64..a3e0f8409ecd3d32a03b799e4ad1fe21c90f2114 100644
--- a/docs/models.md
+++ b/docs/models.md
@@ -2,4 +2,7 @@
 
 The `qim3d` library aims to ease the creation of ML models for volumetric images
 
-::: qim3d.ml.models._unet
\ No newline at end of file
+::: qim3d.ml.models
+    options:
+        members:
+            - UNet
\ No newline at end of file
diff --git a/qim3d/viz/colormaps/_segmentation.py b/qim3d/viz/colormaps/_segmentation.py
index b6ee2f8834b39eaf166da8f1471d9e36d44c2485..cf96e5b7cd1fb3106af548519381a8bf178f1d06 100644
--- a/qim3d/viz/colormaps/_segmentation.py
+++ b/qim3d/viz/colormaps/_segmentation.py
@@ -61,10 +61,10 @@ def segmentation(
         ```python
         import qim3d
 
-        cmap_bright = qim3d.viz.colormaps.objects(num_labels=100, style = 'bright', first_color_background=True, background_color="black", min_dist=0.7)
-        cmap_soft = qim3d.viz.colormaps.objects(num_labels=100, style = 'soft', first_color_background=True, background_color="black", min_dist=0.2)
-        cmap_earth = qim3d.viz.colormaps.objects(num_labels=100, style = 'earth', first_color_background=True, background_color="black", min_dist=0.8)
-        cmap_ocean = qim3d.viz.colormaps.objects(num_labels=100, style = 'ocean', first_color_background=True, background_color="black", min_dist=0.9)
+        cmap_bright = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'bright', first_color_background=True, background_color="black", min_dist=0.7)
+        cmap_soft = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'soft', first_color_background=True, background_color="black", min_dist=0.2)
+        cmap_earth = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'earth', first_color_background=True, background_color="black", min_dist=0.8)
+        cmap_ocean = qim3d.viz.colormaps.segmentation(num_labels=100, style = 'ocean', first_color_background=True, background_color="black", min_dist=0.9)
 
         display(cmap_bright)
         display(cmap_soft)
@@ -77,11 +77,11 @@ def segmentation(
         import qim3d
 
         vol = qim3d.examples.cement_128x128x128
-        binary = qim3d.processing.filters.gaussian(vol, sigma = 2) < 60
-        labeled_volume, num_labels = qim3d.processing.operations.watershed(binary)
+        binary = qim3d.filters.gaussian(vol, sigma = 2) < 60
+        labeled_volume, num_labels = qim3d.segmentation.watershed(binary)
 
-        cmap = qim3d.viz.colormaps.objects(num_labels, style = 'bright')
-        qim3d.viz.slicer(labeled_volume, axis = 1, cmap=cmap)
+        cmap = qim3d.viz.colormaps.segmentation(num_labels, style = 'bright')
+        qim3d.viz.slicer(labeled_volume, slice_axis = 1, color_map=cmap)
         ```
         ![colormap objects](assets/screenshots/viz-colormaps-objects.gif)