Added test case for different scenarios