Add the functionality to return the CPU time