Fix docstrings (thanks @manishtomar :))