Code clean-up using flake8.