Add disconnect feature