Bug fixed

Fixed first bug
Merge pull request #2 from ohadbeny/Fix-bug

Bug fixed