Add time diff counting