I got 2 test cases right but other 2 are not correct. Please comment and correct the code