Update optimization_interface.py