Update benchmark code