Typo
Update demo.py
Revert cosmetics