Update docs
Merge pull request #333 from dungpa/cleanup

Update docs