Update 2a.py