Fix with flake8 code to PEP8