Edit test2.py