Added a test for Cox Stratification