fixing issue #3
Merge pull request #4 from bence-eros/master

Fixing issue #3