Update HW 1 Code