add timing to python script