Updated Sampler to allow for class imbalance

Updated sampler to allow for arbitrary number of classes and class
imbalance by specifying the proportion of data points labeled by a
particular class