Updated benchmark script