pull out groupByKey