Use stratified sampling