Add ok_group function for grouping tests