code style improvement
Merge pull request #4 from lisztli/code-styling

code style improvement