Add websocket support