Fix incorrect docstring
Add module docstring