Changing square loss to match paper