made the code PEP-8 compatible