add stratified sampling