Added an update handler