Added class docstrings
Cleaned up most long lines (PEP8)