step 2 part1 HW4. OK