Fix login screen UX issues