PEP8 cleanup
PEP8 cleanup