Fix flake8 offences