+ updated the code for the new dataset
+ stratified analysis