Fixing up some docstrings, line length, and imports