diff --git a/.github/workflows/train.yaml b/.github/workflows/train.yaml
index bc1f8001..bb18bea7 100644
--- a/.github/workflows/train.yaml
+++ b/.github/workflows/train.yaml
@@ -14,9 +14,10 @@ jobs:
           dvc repro 
 
           git fetch --prune
+          echo "## Metrics: Workspace vs. Main"
           dvc metrics diff --show-md master > report.md
 
           # Add figure to the report
-          echo "## Validating results by region"
+          echo "## Validating results by region" >> report.md
           cml-publish by_region.png --md >> report.md
-          cml-send-comment report.md
\ No newline at end of file
+          cml-send-comment report.md
diff --git a/train.py b/train.py
index e37f0eb5..7f2c76cb 100644
--- a/train.py
+++ b/train.py
@@ -1,6 +1,6 @@
 import pandas as pd 
 import numpy as np
-from sklearn.linear_model import LogisticRegression
+from sklearn.discriminant_analysis import QuadraticDiscriminantAnalysis
 from sklearn import preprocessing
 from sklearn.model_selection import cross_val_predict
 from sklearn.metrics import confusion_matrix
@@ -28,7 +28,7 @@
 
 
 # Linear model
-clf = LogisticRegression()
+clf = QuadraticDiscriminantAnalysis()
 yhat = cross_val_predict(clf, X, y, cv=5)
 
 acc = np.mean(yhat==y)