small fixes
Merge pull request #32 from Studyokee/polish

small fixes