updates:
 - stratified data trimming
 - fix augmentation bug
 - add post processing on augmentations
