Updated docstring formatting
Cleared up logical structure