Add proper benchmarking code, previous one was shit