Add stopwatch function