PEP-8 fixes