diff --git a/atem_core/__init__.py b/atem_core/__init__.py
new file mode 100644
index 0000000..af46af8
--- /dev/null
+++ b/atem_core/__init__.py
@@ -0,0 +1,2 @@
+from .adaptive import *
+from .core import *
\ No newline at end of file
diff --git a/atem_core/adaptive/__init__.py b/atem_core/adaptive/__init__.py
new file mode 100644
index 0000000..85fd970
--- /dev/null
+++ b/atem_core/adaptive/__init__.py
@@ -0,0 +1,6 @@
+from .test_interp import interpret_model
+
+__all__ = [
+    "interpret_model",
+
+]
\ No newline at end of file
diff --git a/atem_core/adaptive/train.py b/atem_core/adaptive/train.py
index 3c27189..76d0ba8 100644
--- a/atem_core/adaptive/train.py
+++ b/atem_core/adaptive/train.py
@@ -26,4 +26,9 @@
 converter = tf.lite.TFLiteConverter.from_keras_model(model)
 tflite_model = converter.convert()
 with open("adaptive_model.tflite", "wb") as f:
-    f.write(tflite_model)
\ No newline at end of file
+    f.write(tflite_model)
+
+
+
+
+
diff --git a/atem_core/core/__init__.py b/atem_core/core/__init__.py
new file mode 100644
index 0000000..315d70e
--- /dev/null
+++ b/atem_core/core/__init__.py
@@ -0,0 +1,2 @@
+from .task_gen import generate_task_data
+from .interp_test import *
\ No newline at end of file