still improving the test script