UI improvements in register and login view