Change from comment to docstring
Update module docstrings