Clean up the registration UI and flow