Fixed flake8 complaints.