Add ability to disconnect