fixed minor feature issue. Fixes #2
Merge pull request #3 from irkanu/master

fixed minor feature issue. Fixes #2