Adding some comments to help understand the demo code better.
See #7