add another test for linear regression