cleanup: make code follow PEP8