Minor changes to benchmark loop