Add update function
Add update function