Update untitled.py