modified test.py