added "Enter number of iterations" prompt