Improve Websocket example