diff --git a/regression.py b/regression.py
index 88902cf..0841cc1 100755
--- a/regression.py
+++ b/regression.py
@@ -87,7 +87,8 @@ def cost(self, w, X, Y, lamb):
     Y_test = np.matrix(Y_test).T
     
     reg = LinearRegression()
-    reg.train(X_train, Y_train) 
+    reg.train(X_train, Y_train)  #you can use regularization parameter to reduce overfitting.
+
     reg.test(X_train, Y_train) 
     reg.test(X_test, Y_test)