Add flattenMap util function