Make code PEP 8 compliant