Improve PersistentMap API