Updated docstrings and comments

Updated docstrings and comments to follow style guidelines and google
docstring suggested format.