removed unused import, pep 8 fix