Add timings to unit tests