add some unit tests
added unit test