Add function to return posterior covariance matrix of random effects