Add more unit tests to increase overall test coverage