Slight edit of the benchmark script