Fix a few typos

Mostly in docstring

Tests: