removed minor flake8 warnings