LinearRegression class changed