Add step2 functionality
Add step2 functionality