doc changes
minor changes to demo.py