Added websocket code