another for playing with class weights and stratified sampling