Update DEMO.py

Full demo