Fix formatting to pass flake8 tests