Pass num_iters also