test.py modified