Updated test cases

All test cases passed