modify python code style