Updated the benchmark script