updated test2.py