Update round1.py