Removing unused import
Flake8