Working on adding groomed jet collections