Unit test cases updated