Added polish to login flow