cleanup: make the code follow PEP8