Enhancement/doc comments fixes #8

Add doc comments to current source. fixes #8