refactor: cleanup benchmarking code