PEP8 cleanup of #31

PEP8 cleanup