Fix all the things (try demo.py)