Update the section about the average loss