Updates to test_script.rb