Finish the assignment with all test pass