Fix issue #3
Merge pull request #4 from Eyyub/master

Fix issue #3