Remove unused import
PEP 8 fixes