Update HW1_optimize_sortino.py