Update test_ex2.py