Update example to match new format
Fix pep8 violations