add unit test
updates