fixed bug #4
Merge pull request #20 from samuel-yeom/master

fixed bug #4