updated unit tests
updated unit tests