Update benchmark script to use new API