adjust Demo.py