PEP-8 fixes.

Fixed some docstring placement and a * import.