diff --git a/README.md b/README.md
index 003cb75..e9d30bf 100644
--- a/README.md
+++ b/README.md
@@ -39,3 +39,8 @@ print(out.shape)
 
 # License
 MIT
+
+# Todo
+- [ ] Add paper link
+- [ ] Add citation bibtex
+- [ ] cleanup
\ No newline at end of file
diff --git a/simba_torch/__init__.py b/simba_torch/__init__.py
index e69de29..13ce743 100644
--- a/simba_torch/__init__.py
+++ b/simba_torch/__init__.py
@@ -0,0 +1,3 @@
+from simba_torch.main import Simba
+
+__all__ = ["Simba"]
\ No newline at end of file