Updated test script a little bit.