cleaned up code
Merge pull request #2 from janeplural/patch-2

cleaned up code