Add MSE training cost