updare code.py