Changed the way the progress callback works