Tweaks to iteration exercise