Added some disconnect button functionality.