diff --git a/complexipy/__init__.py b/complexipy/__init__.py
index e69de29..2ace38b 100644
--- a/complexipy/__init__.py
+++ b/complexipy/__init__.py
@@ -0,0 +1,18 @@
+from complexipy.rust import (
+    CodeComplexity,
+    FileComplexity,
+    FunctionComplexity,
+)
+
+from .main import (
+    code_complexity,
+    file_complexity,
+)
+
+__all__ = [
+    "CodeComplexity",
+    "FileComplexity",
+    "FunctionComplexity",
+    "code_complexity",
+    "file_complexity",
+]
diff --git a/tests/main.py b/tests/main.py
index 3002204..c64061a 100644
--- a/tests/main.py
+++ b/tests/main.py
@@ -2,7 +2,10 @@
 import unittest
 from pathlib import Path
 
-from complexipy.main import file_complexity, code_complexity
+from complexipy import (
+    file_complexity,
+    code_complexity,
+)
 
 
 class TestFiles(unittest.TestCase):