Update test_2.py