Improvements to login screen