HW2 Part 1 initial code change