PEP 8 code fixes