add demo.py