fixed issue 4
Merge pull request #20 from zboh12/master

fixed issue 4