HW4 solution