Incorporate review feedback

- update docs