Reworking the websocket ex into a chat