HW2 update