Added edit text back into congrats step when needed. Closes #164