Fix bug
Merge pull request #11 from Jat15/patch-1

Fix bug