clean up repeated demo code
Merge pull request #7 from caranicas/democlass

clean up repeated demo code