made code PEP8 compliant