Added a test for Cox Stratification
Finish output test for stratified cox
Fix input tests
Fix the data loading problem