remove unused import and PEP8.