Add progress tracking functionality