Updated the example to work with the new WebSocket decorator