Add demo.py, update readme