Add heterozygosity to snp.py