diff --git a/models/model.pkl b/models/model.pkl
index b33a4ffb..6c55e7e0 100644
Binary files a/models/model.pkl and b/models/model.pkl differ
diff --git a/requirements.txt b/requirements.txt
index c36e9f95..86b7f277 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ pytz==2020.1
 scipy==1.5.2
 six==1.15.0
 threadpoolctl==2.1.0
+scikit-learn
diff --git a/train.py b/train.py
index 2dcb69be..7662da2f 100644
--- a/train.py
+++ b/train.py
@@ -1,12 +1,12 @@
 from sklearn.datasets import make_regression
-from sklearn.linear_model import LinearRegression
+from sklearn.linear_model import Lasso
 import pickle
 import os
 
-X, y = make_regression(10000,n_features = 10)
+X, y = make_regression(10000,n_features = 11)
 
 # Train a model
-reg = LinearRegression().fit(X, y.ravel())
+reg = Lasso().fit(X, y.ravel())
 # Print out training r2
 print(reg.score(X,y.ravel() ))