Improve the benchmark for the version 1