working on PEP 8 formatting