Put logout function back!