updated the main testing script