Add some new websocket features