add ui for step2 and 3