Add timing code