COSMIT remove unused imports, pep8