Add number of iterations as input into the script