Fix up notebook OptimizationDemo1