removed unused import + PEP 8