Update optimizer.py