fixing bug 3
Merge pull request #4 from pinguxx/master

Fixing bug #3
thanks