added a unit test for linear regression