Updated README.md to include demo. Also added some brief comments / formatting fixes to the source file