Add some timing code