Tidy up benchmark script