minor changes to iterations part