Add simple websocket function