Adjust the code layout
Add `demo.py`
Add README