How to check the running time of program in python