diff --git a/src/database/__init__.py b/src/database/__init__.py
index 4083b77..e790a76 100644
--- a/src/database/__init__.py
+++ b/src/database/__init__.py
@@ -1,5 +1,11 @@
 from .classes import (
     ModelOutputs,
-    VisualCommunication
+    VisualCommunication,
+    NoDocumentFoundException
+)
+from .database import connect
+from .utils import (
+    total_documents,
+    total_annotated,
+    get_visual_communication
 )
-from .database import connect
\ No newline at end of file