Update iteration.py