Fix flake8 whitespace errors