add some models and websocket logic