PEP 8 fixes