python code update