Add websocket controll