Update optimization.py

Removed comments