Python Code For Optimization Problem