Update demo_01.py