Add CovarianceCost (trace of the covariance). Try different observation models.