Tweaks to benchmark script