update test2.py