Updating test_script.py, seems things got messy