Added some unit-test code.