change test_lreg to use code im example.linear_regression