Updated optimization notebook