Fixed Linear Regression, added new features