modify opti_HW4_q2.py and add result_q1 and result_q2