Add update method to the heap