Undoing change to squareLoss, it doesn't work