Update HW4 - 2