Update demo.py for python-demo