modify code.py