added assertions in diagonal gaussian
removed np.diag calls in DiagonalGaussian sumsq (bug fix)