Added test cases for the change