Minor update to login functionality