Tweak benchmark script