Add websocket suppport