add some more unit test