Add test grouping