Made code PEP 8 compliant