Made a slight update to the main test script.