Small fixes
Merge pull request #11 from MichalPaszkiewicz/master

Small fixes