Add some documentation

Also change to use more WebSocketResult