remove not used import. refactoring based on pep8