add comment
add doc comment
update imports
Merge pull request #139 from wongak/docs_cleanup

Docs and cleanup