Fixing flake8 violations F401