Proof of concept using demo.py