Change update method to persist changes