forgot to add optimize.py