Add websocket handling.