Update demo.py

Better comments