HW1 code