Add HW4 solution