Added save for update function