Add more documentation

Also fix a change in WebSocketClient API