properly update output pcollections from expanded transforms

currently output pcollections from expanded transforms are ignored. we need to properly update output pcollections when it's returned to the caller of expansion service.