adding more unit tests.