Fixing HW4_2c