add unit test
minor changes