add step2 code