fixed issue #4
Merge pull request #7 from joecollins/issue#3

fixed issue #4