Remove unused import and PEP8 cleanup.