Fix python styling