Add in buttons for step 2. Add in shell for step 3.