PEP8-ify python code