Update HW2_final