Update python code.