UI improvements and progress bar