Add some major functionality to websocket