Update demo1.py