Add option to change iterations