Added more functionality to progress bar