Still trying to fix linear regression