Added disconnect functionality