Added some more text to the average loss section