Cleaned up some docstrings and formatting.