Python: code style fixes

I believe I did not screw anything up