Improve unit test script for test driven development.