Improve benchmarking script.