Merge pull request #362 from mhils/patch-2

Fix docstring types