Added an on update finished callback