Updated optimization section