diff --git a/main.py b/main.py
index e07ec5f..df5fba2 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,8 @@
 sample data, and it's like generating droids.
 """
 
+import pyspark
+
 if __name__ == "__main__":
     pass
 
diff --git a/training/training_scheduler.py b/training/training_scheduler.py
index b8cb5ed..4c49c04 100644
--- a/training/training_scheduler.py
+++ b/training/training_scheduler.py
@@ -8,4 +8,10 @@ def __init__(self, hyperparameters, model_directory, model):
     def train(self):
         return
 
+    def visualize_model(self):
+        return
+
+    def plot_training_progress(self):
+        return
+
 # EOF