Remove unused import
PEP8 cleanups