add timing code back