[TASK] update the demo script