Add controls for logout