Fix py files to align with flake8