Start to add unit tests.