Websocket functionality addition - first version