Add demo.py