Add HW6 solution