diff --git a/codegen/__main__.py b/codegen/__main__.py
index cd9ac48..fae93c7 100644
--- a/codegen/__main__.py
+++ b/codegen/__main__.py
@@ -1,6 +1,6 @@
-def main():
-    pass
-
+from .codegen import test_str8, test_doubly_linked_list, test_typed_array
 
 if __name__ == "__main__":
-    main()
+    test_str8()
+    test_typed_array()
+    test_doubly_linked_list()
diff --git a/codegen/codegen_example.py b/codegen/codegen.py
similarity index 100%
rename from codegen/codegen_example.py
rename to codegen/codegen.py
diff --git a/codegen/dev_main.py b/codegen/dev_main.py
deleted file mode 100644
index d1078da..0000000
--- a/codegen/dev_main.py
+++ /dev/null
@@ -1,6 +0,0 @@
-from .codegen_example import test_str8, test_doubly_linked_list, test_typed_array
-
-if __name__ == "__main__":
-    test_str8()
-    test_typed_array()
-    test_doubly_linked_list()
diff --git a/run b/run
index 19e8e9d..6d8984c 100755
--- a/run
+++ b/run
@@ -1,3 +1,3 @@
 #!/bin/bash
 
-python3 -m codegen.dev_main
+python3 -m codegen