Add model_iterator argument