Add test2.py