flake8 cleanup (removed unused import)