Add websocket support code