Improve login box