add python code zhushi