Allow user to disconnect from websocket

Separated this into another branch