Adding timing code to python client