Improved the login/ logout UI