Improve benchmark script.