Update python-code