Login and registration UI fixes