Small change to logout