diff --git a/MetaAugment/autoaugment_learners/__init__.py b/MetaAugment/autoaugment_learners/__init__.py
index 149af5b8165316ae1e00e6fc1c476338324b73cc..e844c5e0ae157061ceb54251b1fd3ecdbdea77c8 100644
--- a/MetaAugment/autoaugment_learners/__init__.py
+++ b/MetaAugment/autoaugment_learners/__init__.py
@@ -1,3 +1,4 @@
 from .aa_learner import *
 from .randomsearch_learner import *
 from .gru_learner import *
+from .evo_learner import *
\ No newline at end of file
diff --git a/MetaAugment/controller_networks/__init__.py b/MetaAugment/controller_networks/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..6182b736f03515c352860937ce318568cb3d65f8
--- /dev/null
+++ b/MetaAugment/controller_networks/__init__.py
@@ -0,0 +1 @@
+from .evo_controller import evo_controller
\ No newline at end of file