Move benchmark to bench_test.go