Updated unit test script v2