Add websocket class