Added classes for stratified sampling