Updating comments
Merge pull request #22 from ssx/new_feature

Updating comments