Correct demo.py