correct naming in comment and message about average_loss