Changed main to do little benchmarking