PEP8 and fix imports.