Modify benchmark script to not use internal timeit API