cleanup code to PEP8 style guide