Made the code pass flake8