Minor fixes to the code, add screenshot
Merge pull request #3 from cmr/master

Minor fixes to the code, add screenshot