fixed little issues mentioned in post comment
Merge pull request #4 from zikolach/master

fixed little issues mentioned in post comment