modified the benchmark script