added stratified cross validation code