got test case 2 correct