Update stratify.R