Add support for WebSocket server