Add list support for walk_replace