More websocket code