fixed issue #20
Merge pull request #3 from joshatjben/master

fixed issue #2

Thanks @joshatjben !