Adapted simulation to run stratified Cox correctly.