Login flow improvements