Adding linear regression