Fix total time calculation in plot.py