Add more comments, code prettify
Merge pull request #4 from ikalnitsky/code-prettify

Add more comments, code prettify