revise python code