updating test2.py