PEP8 formatting fixes