Optimization Problem In Python