diff --git a/doc/reference/data_structures.rst b/doc/reference/data_structures.rst
index 23be5df..e031397 100644
--- a/doc/reference/data_structures.rst
+++ b/doc/reference/data_structures.rst
@@ -4,6 +4,12 @@ Data structures
 
 .. currentmodule:: colabseg
 
+ColabSegData
+~~~~~~~~~~~~
+.. autosummary::
+   :toctree: api/
+
+   new_gui_functions.ColabSegData
 
 ImageIO
 ~~~~~~~
diff --git a/doc/reference/gui_interactions.rst b/doc/reference/gui_interactions.rst
index 921135c..980d716 100644
--- a/doc/reference/gui_interactions.rst
+++ b/doc/reference/gui_interactions.rst
@@ -1,17 +1,20 @@
 GUI Interactions
 ================
 
+.. currentmodule:: colabseg
+
+
 Segmentation
 ~~~~~~~~~~~~
 .. autosummary::
    :toctree: api/
 
-   colabseg.segmentation_gui.JupyterFramework
-   colabseg.tensorvoting_wrapper.generate_tensor_voting_gui
+   segmentation_gui.JupyterFramework
+   tensorvoting_wrapper.generate_tensor_voting_gui
 
 Controls
 ~~~~~~~~
 .. autosummary::
    :toctree: api/
 
-   colabseg.py3dmol_controls.seg_visualization
+   py3dmol_controls.seg_visualization