Fixed some flake8 warnings.