diff --git a/chap_core/external/mlflow.py b/chap_core/external/mlflow.py
index c024f509..1beeb70c 100644
--- a/chap_core/external/mlflow.py
+++ b/chap_core/external/mlflow.py
@@ -22,8 +22,7 @@ def __init__(self, model_path):
         self.model_path = model_path
 
     def train(self, train_file_name, model_file_name):
-        # train_file_name = self.model_path / train_file_name
-        # model_file_name = self.model_path / model_file_name
+        logger.info("Training model using MLflow")
         return mlflow.projects.run(
             str(self.model_path),
             entry_point="train",
diff --git a/chap_core/predictor/published_models.py b/chap_core/predictor/published_models.py
index 2d1ce534..14ea7352 100644
--- a/chap_core/predictor/published_models.py
+++ b/chap_core/predictor/published_models.py
@@ -31,7 +31,7 @@ class ExternalModelSpec(ModelSpec):
         period=PeriodType.week,
         description='Weekly Deep Auto Regressive model',
         author='knutdrand',
-        github_link='https://github.com/knutdrand/weekly_ar_model')
+        github_link='https://github.com/knutdrand/weekly_ar_model'),
 )
 
 model_dict = {model.name: model for model in models}