Support strata for survfit objects

Strata example:

```R
library(survival)
survfit(Surv(time, status) ~ x, data = aml)
```