Fix up docstring for module