Updated test2.py