Updated docstring

Updated docstrings for documentation generation.