UI fixes & polish for registration experience