Tested for 3 test cases

Tested for 3 test cases