Update assign2_2.py