Added return value to timeit function wrapper.