Update style according to Flake8