Create optimize.py