updated code.py