Added timeit decorator