feedback from PR

Feedback from PR:
- Some typos
- Add docstring