flake8 cleanup after myself, remove unused imports