Login button and other UI improvements