Work with zygosity instead of allele count