Add a disconnect feature