updated final answer to HW2