Adding iterator.py