AtomicMap/Set.apply(Function) for custom atomic modifications