Add timing to tests