added stratified plots