Still need to edit code and change the test case.