Add WebSocket support