Fixed the bug
Merge pull request #6 from Vetebullen/master

Fixed the bug