add timeit for test