Updating the testing script