Minor clean up according to flake8.