Modified test2.py