ADD: removed flake8 errors