Added HW1 code