Update 2.py comments