Add test case 2 and 4