improve benchmark script