updated step 2 layout