fixed docstring
Merge pull request #4 from wujiang/patch-1

fixed docstring