added unit test
updated unit test