add progress bar for both treatments