Adding option for warmup iterations