update websocket API
update docs for new API