Fixed some flake8 warnings