Update benchmark script for shorten the run time.