Made the __main__ demo make sense (Fixes #82)
Merge pull request #127 from helenst/reusing-code-fix-demo

Made the __main__ demo make sense (Fixes #82)