More python style

Just a style fix