Update for flake8 compliance.
Update for flake8 compliance.