Starting to add unit tests.