Take a callback when logging out too