made new additions flake8 compliant