Fix docstrings

Fixes #25 and #26