Prepare code for benchmarking