Added compliance with flake8