COSMIT: Made calculation of mse a bit clearer