Reformatted python code to be PEP8 compliant

See title, no other changes