Heritability calculation