Loss function (40/42)

Loss function