Fix flake8 whitespace issues