Using timeit to profile Python code