updated to include counts for each sample