WIP add websocket stuff