diff --git a/MaCh3PythonUtils/machine_learning/fml_interface.py b/MaCh3PythonUtils/machine_learning/fml_interface.py
index 6a22cd6..3b9029f 100644
--- a/MaCh3PythonUtils/machine_learning/fml_interface.py
+++ b/MaCh3PythonUtils/machine_learning/fml_interface.py
@@ -9,7 +9,6 @@
 from sklearn import metrics
 from sklearn.preprocessing import StandardScaler
 import matplotlib.pyplot as plt
-import scipy.stats as stats
 import numpy as np
 
 import pickle
diff --git a/MaCh3PythonUtils/machine_learning/ml_factory.py b/MaCh3PythonUtils/machine_learning/ml_factory.py
index 2f1b236..c48411b 100644
--- a/MaCh3PythonUtils/machine_learning/ml_factory.py
+++ b/MaCh3PythonUtils/machine_learning/ml_factory.py
@@ -51,13 +51,13 @@ def __setup_package_factory(self, package: str, algorithm: str, **kwargs):
             
         return self.__IMPLEMENTED_ALGORITHMS[package][algorithm](*kwargs)
 
-    def setup_scikit_model(self, algorithm: str, **kwargs)->SciKitInterface:
+    def make_scikit_model(self, algorithm: str, **kwargs)->SciKitInterface:
         # Simple wrapper for scikit packages
         interface = SciKitInterface(self._chain, self._prediction_variable)
         interface.add_model(self.__setup_package_factory(package="scikit", algorithm=algorithm, **kwargs))
         return interface
     
-    def setup_tensorflow_model(self, algorithm: str,  **kwargs):
+    def make_tensorflow_model(self, algorithm: str,  **kwargs):
         interface = TfInterface(self._chain, self._prediction_variable)
         
         interface.add_model(self.__setup_package_factory(package="tensorflow", algorithm=algorithm))
diff --git a/configs/tensorflow_config.yml b/configs/tensorflow_config.yml
index 5fbf1b6..e9c3491 100644
--- a/configs/tensorflow_config.yml
+++ b/configs/tensorflow_config.yml
@@ -44,4 +44,3 @@ MLSettings:
 
         FitSettings:
             batch_size: 100
-            epochs: 200
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 22123b5..90c849f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,4 +12,4 @@ arviz
 mplhep
 tqdm
 numba
-jinja2
\ No newline at end of file
+jinja2