Use linear regression on train data
Minor fix - Add Test data