Fix docstrings, specifically for checking types

[bobleesj]

Please see: #120 (comment)

Maybe we could use mypy for static type checking

[john-halloran]

Closed by #187.

[sbillinge]

closed per @john-halloran