Updated the benchmark script.