Could the loss function be any better??