Updated test.py