Create optimization.py