python style fix