Update portfolio-optimization.py