PEP-8-ify code