PEP8 edits

not sure if it's better, but it is compliant