changed to linear regression