Update 2_1.py