PEP 8 corrections