update demo.py