code cleanup, PEP8