Adding a few unit test.