Loss function (20/42)

Loss function