Update HW_5 again