Make benchmark script easier to use.