Updated test cases
update unit tests