Fix #2

Fix #2
Merge pull request #4 from MikeMedvedev/master

Fix #2