Changing square loss to be for one x/z pair at a time