Added choice to specify iterations