PEP8 formatting cleanup