fix code for step1