bug fix in code

The code has been fixed
Merge pull request #31 from mayank8888/patch-1

bug fix in code