Adding the update part