Update test-script.py