Clean up benchmark script