Add timing code. Currently commented out.