Cleanup & code syntax fix

Signed-off-by: Ivan Bilych <ivan.bilych@gmail.com>
Merge pull request #2 from ivanbilych/fix#1

Cleanup & code syntax fix