Update HW2 answer