Update to iteration class