Add styling for login flow