Now the websocket is class