Added the optimization problem