Small code cleanup
Merge pull request #10 from CodeInsider/patch-3

Small code cleanup