PEP-8 formatting fixes