add unit test
update unit test