Adds more unit tests.