Changed test.py