add ability to specific number of samples and make sure when sampling with stratification get total number requested back