helpful comments

Added helpful comments to the demo.py