Add keyBy & groupBy functions