Cleaned up docstrings, comments, and spacing