Style fix - Docstrings

Updated the docstrings of the project