Simplified code in ooc groupby demo