Refactored code to comply own standard.

fixes #11 
Maybe a block comment?