minor bug fixed
Merge pull request #5 from monttj/master

minor bug fixed