Added linear regression function