diff --git a/core/__init__.py b/core/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/core/database/__init__.py b/core/database/__init__.py
index 2ae7dd7..238806b 100644
--- a/core/database/__init__.py
+++ b/core/database/__init__.py
@@ -3,10 +3,9 @@ from __future__ import annotations
 from .base import connect
 from .classes import NoDocumentFoundException
 from .classes import VisualCommunication
+from .utils import count_documents
 from .utils import get_visual_communication
 from .utils import list_names
-from .utils import total_annotated
-from .utils import total_documents
 from .utils import upsert_annotations
 from .utils import upsert_predictions
 from .utils import upsert_visual_communication