Add linear mix model to impute yield