Add websocket part