Added new test cases.
Test case fix