diff --git a/iss/models/__init__.py b/iss/models/__init__.py
index f6f4288..456ba5b 100644
--- a/iss/models/__init__.py
+++ b/iss/models/__init__.py
@@ -5,3 +5,5 @@ from .SimpleConvAutoEncoder import SimpleConvAutoEncoder
 from .SimpleAutoEncoder import SimpleAutoEncoder
 from .VariationalAutoEncoder import VarAutoEncoder
 from .VariationalConvAutoEncoder import VarConvAutoEncoder
+from .ModelTrainer import ModelTrainer
+from .DataLoader import ImageDataGeneratorWrapper