COSMIT pep8, removing unused imports