hw1 step2 - part A