Updating test.py