add demo.py file