Add update method