estimate heritability only when fitting a model