Add progress indication for training