Update Q1_2.py