homework 4 FIXED