Made code PEP8 compliant.