Enhancement/doc comments

Add doc comments to current source