redo formatting

to match python style