Add support for progress bar