add test2.py