notebooks updated after changes in flatnessLoss