Making code PEP8 compliant