HW4_2c fixed