hw4 need to finish step2