diff --git a/classifier/__init__.py b/classifier/__init__.py
index cf5c18a..e53691d 100644
--- a/classifier/__init__.py
+++ b/classifier/__init__.py
@@ -1,5 +1,7 @@
 from classifier import in_training, print_answers
 from classifier import get_pipe, test_pipe, get_training_threads
-from training import train_single_thread
 from classifier import store_training_data
+
+from training import train_single_thread
+
 from prediction import predict_threads
diff --git a/storage/__init__.py b/storage/__init__.py
index 2a9def7..e6ffa63 100644
--- a/storage/__init__.py
+++ b/storage/__init__.py
@@ -1,5 +1,8 @@
+# SQL Database session
 from database import db_session, init_db
-from mail_model import Mail
-from thread_model import MailThread
 from database import Base
 metadata=Base.metadata
+
+# Two main objects: Mail & MailThread
+from mail_model import Mail
+from thread_model import MailThread