Added some unit tests.