Working on better progress bar