STY: PEP8, white-space, unused import.

Some minor cleanups.