adding some more unit tests