Working linear regression homework for class