Check my code, only one test case passed