Fix bug 2
Merge pull request #12 from Jat15/patch-2

Fix bug 2