Update demo.py