Small change to logout functionality