Update question_3.1.py