Fix typos
Add docstrings