Get linear regression to work