cleaned the benchmark code