Add saving of root set